Spaces:
Paused
Paused
Update app.py
Browse files
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, '
|
| 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
|
|
|
|
| 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,
|
| 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")
|