MrUtakata commited on
Commit
72996f9
·
verified ·
1 Parent(s): bf47bf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ def preprocess_input(row_values):
36
  input_df['byte_ratio'] = input_df['sbytes'] / (input_df['dbytes'] + 1)
37
  input_df['pkt_ratio'] = input_df['Spkts'] / (input_df['Dpkts'] + 1)
38
 
39
- # Drop unused or label columns
40
- input_df = input_df.drop(columns=features_to_drop + ['attack_cat', 'Label'], errors='ignore')
41
 
42
  return input_df
43
 
 
36
  input_df['byte_ratio'] = input_df['sbytes'] / (input_df['dbytes'] + 1)
37
  input_df['pkt_ratio'] = input_df['Spkts'] / (input_df['Dpkts'] + 1)
38
 
39
+ # Fix: convert features_to_drop to list before adding with another list
40
+ input_df = input_df.drop(columns=list(features_to_drop) + ['attack_cat', 'Label'], errors='ignore')
41
 
42
  return input_df
43