initial commit

Sniffer_AI.py

@@ -9,9 +9,9 @@ from sklearn.tree import DecisionTreeClassifier
 
 # Load the saved models
 rf_model = joblib.load('rf_model.pkl')
-dt_model = joblib.load('dt_model.pkl')
-bagging_model = joblib.load('bagging_model.pkl')
-ada_model = joblib.load('ada_model.pkl')
+dt_model = joblib.load('decision_tree_model.pkl')
+bagging_model = joblib.load('model_bagging.pkl')
+ada_model = joblib.load('model_adaboost.pkl')
 
 class_labels = {
     0: "normal",
@@ -34,11 +34,11 @@ def detect_intrusion(features, model_choice="Random Forest"):
     if model_choice == "Random Forest":
         model = rf_model
     elif model_choice == "Decision Tree":
-        model = dt_model
+        model = decision_tree_model
     elif model_choice == "Bagging Classifier":
-        model = bagging_model
+        model = model_bagging
     elif model_choice == "AdaBoost Classifier":
-        model = ada_model
+        model = model_adaboost
     else:
         return "Invalid model choice!"