Mekam commited on
Commit
bdb8a39
·
1 Parent(s): fdcd714

fix(prediction): delete label from required column

Browse files
src/controllers/prediction_controller.py CHANGED
@@ -11,7 +11,7 @@ REQUIRED_COLUMNS = [
11
  "psh_flag_number", "ack_flag_number", "ece_flag_number",
12
  "cwr_flag_number", "ack_count", "syn_count", "fin_count",
13
  "rst_count", "TCP", "UDP", "Tot sum", "Min", "Max", "AVG",
14
- "Std", "Tot size", "IAT", "Number", "Variance", "label"
15
  ]
16
 
17
  def global_prediction_on_csv(file: UploadFile):
 
11
  "psh_flag_number", "ack_flag_number", "ece_flag_number",
12
  "cwr_flag_number", "ack_count", "syn_count", "fin_count",
13
  "rst_count", "TCP", "UDP", "Tot sum", "Min", "Max", "AVG",
14
+ "Std", "Tot size", "IAT", "Number", "Variance"
15
  ]
16
 
17
  def global_prediction_on_csv(file: UploadFile):