tejani commited on
Commit
8729e50
·
verified ·
1 Parent(s): ed57794

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -21,7 +21,7 @@ def load_and_preprocess_data(file):
21
  data = pd.read_csv(file.name)
22
 
23
  # Convert suits and ranks to numerical values
24
- suit_order = {'spades': 0, 'hearts': 1, 'logs': 2, 'diamonds': 3}
25
  rank_order = {'ace': 0, '2': 1, '3': 2, '4': 3, '5': 4, '6': 5, '7': 6, '8': 7, '9': 8, '10': 9,
26
  'jack': 10, 'queen': 11, 'king': 12}
27
 
@@ -140,7 +140,8 @@ def train_model(file, n_estimators, learning_rate, max_depth, subsample, progres
140
  }
141
 
142
  # Scale features
143
- progress(0.3, desc="Scaling features scaler = StandardScaler()
 
144
  features_scaled = scaler.fit_transform(features)
145
  features_scaled = pd.DataFrame(features_scaled, columns=features.columns)
146
 
@@ -185,7 +186,7 @@ def train_model(file, n_estimators, learning_rate, max_depth, subsample, progres
185
  # Evaluate
186
  y_pred = model.predict(X_test)
187
  accuracy = accuracy_score(y_test, y_pred)
188
- report = classification_report(y_test, y_pred, zero_division=0)
189
  accuracies[target_name] = accuracy
190
 
191
  results.append(f"**{target_name} Results**\n")
 
21
  data = pd.read_csv(file.name)
22
 
23
  # Convert suits and ranks to numerical values
24
+ suit_order = {'spades': 0, 'hearts': 1, 'clubs': 2, 'diamonds': 3}
25
  rank_order = {'ace': 0, '2': 1, '3': 2, '4': 3, '5': 4, '6': 5, '7': 6, '8': 7, '9': 8, '10': 9,
26
  'jack': 10, 'queen': 11, 'king': 12}
27
 
 
140
  }
141
 
142
  # Scale features
143
+ progress(0.3, desc="Scaling features...")
144
+ scaler = StandardScaler()
145
  features_scaled = scaler.fit_transform(features)
146
  features_scaled = pd.DataFrame(features_scaled, columns=features.columns)
147
 
 
186
  # Evaluate
187
  y_pred = model.predict(X_test)
188
  accuracy = accuracy_score(y_test, y_pred)
189
+ report = classification_report(y_test, y_pred, zero<colgroup):
190
  accuracies[target_name] = accuracy
191
 
192
  results.append(f"**{target_name} Results**\n")