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