From d1a9059ae519baa45eb10a1f03e24eb549979603 Mon Sep 17 00:00:00 2001 From: ukfnu <ukfnu@student.kit.edu> Date: Sun, 23 Mar 2025 23:47:46 +0100 Subject: [PATCH] final clean ups --- README.md | 2 ++ code/main.py | 3 +-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 0910552..c20a120 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,7 @@ # Intrusion Detection System with Machine Learning +## It is important to launch this project with the python virtual environment (.venv). If your python interpreter +```is not in the project order, please replace the variable python_executable in the file main.py with the path to your python interpreter. ## Packages needed to be installed: - `scapy` - `duckdb` diff --git a/code/main.py b/code/main.py index 9de166b..d463540 100644 --- a/code/main.py +++ b/code/main.py @@ -8,6 +8,7 @@ import duckdb sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'machine_learning_models'))) base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) +# Path to the Python executable in the virtual environment, please replace if it divers with your python_executable = os.path.join(base_dir, ".venv", "Scripts", "python.exe") from machine_learning_models import utilities as util @@ -21,8 +22,6 @@ df_train, df_test, model_name = util.import_data( model_name=None ) -from machine_learning_models import random_forest - from matplotlib import pyplot as plt from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg -- GitLab