This project implements a simple Intrusion Detection System (IDS) using machine learning. It allows users to upload CSV files containing network traffic data and checks for potential attacks.
- Clone this repository.
- Install the required dependencies:
pip install -r requirements.txt
- Train the model (make sure you have the dataset):
python model/train_model.py
- Run the Streamlit app:
streamlit run app.py
- Open the Streamlit app in your web browser.
- Upload a CSV file containing network traffic data.
- Click the "Check for Attacks" button to analyze the data.
- View the results, including the percentage of detected attacks and a sample of attack records.
app.py
: Main Streamlit applicationpreprocessing/preprocess.py
: Data preprocessing functionsutils/helpers.py
: Utility functionsmodel/train_model.py
: Script for training the machine learning modelmodel/model.joblib
: Saved machine learning model (generated after training)requirements.txt
: Project dependenciesREADME.md
: Project documentation
Make sure to train the model with your specific dataset before using the application. Adjust the preprocessing steps and model parameters as needed for your use case.