feat(predict): select feature for level 1

src/controllers/prediction_controller.py

@@ -93,10 +93,25 @@ def global_prediction_on_csv(file: UploadFile):
         print("drop label column if exists")
 
         # Préparation des features
+        features = [
+            "Header_Length",
+            "Time_To_Live",
+            "Rate",
+            "Tot sum",
+            "Tot size",
+            "Min",
+            "Max",
+            "AVG",
+            "Std",
+            "Variance",
+            "IAT",
+            "Number"
+        ]
+
         if "label" in data.columns:
-            X = data.drop(columns=["label"])
+            X = data[features]
         else:
-            X = data
+            X = data[features]
             
         print("features prepared, starting prediction")