Rthur2003 commited on
Commit
905436c
·
1 Parent(s): 0d5b4e8

feat: adjust Random Forest parameters for improved model performance

Browse files
Files changed (1) hide show
  1. app/training/train_classifier.py +3 -3
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=12,
258
- min_samples_leaf=8,
259
- min_samples_split=10,
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,