This project builds a machine learning model to detect ransomware infections using memory forensic features.
data/: Input dataset CSVmodels/: Saved model, scaler, encoderssrc/: Training and prediction scriptsapp/: Streamlit UIoutput/: Evaluation results like feature plots
- Install requirements:
pip install -r requirements.txt- Train the model:
python src/train_model.py- Predict a sample:
python src/predict_sample.py- Run Streamlit UI:
streamlit run app/streamlit_app.pyIncludes system internals like process/thread counts, handles, psxview flags, dlls, etc.
Total: 58,596 records (29,298 benign, 29,298 malicious) with 58 behavioral features.
Let me know if you want Docker setup or cloud deployment next.
pycache/ *.py[cod]
models/*.pkl
output/.png output/.log
.env .venv/
.streamlit/
.ipynb_checkpoints/