Spaces:
Sleeping
Sleeping
feat: adjust Random Forest parameters for improved model performance
Browse files
app/training/train_classifier.py
CHANGED
|
@@ -254,9 +254,9 @@ def _build_candidates() -> list[tuple[str, Any]]:
|
|
| 254 |
"Random Forest",
|
| 255 |
RandomForestClassifier(
|
| 256 |
n_estimators=200,
|
| 257 |
-
max_depth=
|
| 258 |
-
min_samples_leaf=
|
| 259 |
-
min_samples_split=
|
| 260 |
class_weight="balanced",
|
| 261 |
random_state=42,
|
| 262 |
n_jobs=-1,
|
|
|
|
| 254 |
"Random Forest",
|
| 255 |
RandomForestClassifier(
|
| 256 |
n_estimators=200,
|
| 257 |
+
max_depth=8,
|
| 258 |
+
min_samples_leaf=15,
|
| 259 |
+
min_samples_split=20,
|
| 260 |
class_weight="balanced",
|
| 261 |
random_state=42,
|
| 262 |
n_jobs=-1,
|