r.learn.ml2: change max_features=auto to sqrt (OSGeo#944)

HuidaeCho · Sep 20, 2023 · fb8fbe7 · fb8fbe7
1 parent e76c325
commit fb8fbe7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/raster/r.learn.ml2/r.learn.train/r.learn.train.py b/src/raster/r.learn.ml2/r.learn.train/r.learn.train.py
@@ -424,7 +424,7 @@ def process_param_grid(hyperparams):
     if hyperparams["max_depth"] == 0:
         hyperparams["max_depth"] = None
     if hyperparams["max_features"] == 0:
-        hyperparams["max_features"] = "auto"
+        hyperparams["max_features"] = "sqrt"
     param_grid = {k: v for k, v in param_grid.items() if v is not None}
 
     return hyperparams, param_grid