From fb8fbe7f92061801e51d853f69ffc3e3438b0718 Mon Sep 17 00:00:00 2001 From: Markus Metz <33666869+metzm@users.noreply.github.com> Date: Wed, 20 Sep 2023 13:55:24 +0200 Subject: [PATCH] r.learn.ml2: change max_features=auto to sqrt (#944) --- src/raster/r.learn.ml2/r.learn.train/r.learn.train.py | 2 +- 1 file changed, 1 insertion(+), 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 index a6507d35cb..d5743404eb 100644 --- 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