From f2d2f361f0c5961e698c3de96a40f1725ee2ffb7 Mon Sep 17 00:00:00 2001 From: Tanushree Tunstall Date: Tue, 24 May 2022 07:50:11 +0100 Subject: [PATCH] n_estimators range to include upto a 1000 alex's suggestion --- uq_ml_models_FS/fs_UQ_RF.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/uq_ml_models_FS/fs_UQ_RF.py b/uq_ml_models_FS/fs_UQ_RF.py index fc7706c..c94f495 100644 --- a/uq_ml_models_FS/fs_UQ_RF.py +++ b/uq_ml_models_FS/fs_UQ_RF.py @@ -32,7 +32,7 @@ param_grid_rf = [ # 'clf': [RandomForestClassifier(**rs, **njobs, bootstrap = True, oob_score = True)], 'clf__max_depth': [4, 6, 8, 10, 12, 16, 20, None] , 'clf__class_weight':['balanced','balanced_subsample'] - , 'clf__n_estimators': [10, 25, 50, 100] + , 'clf__n_estimators': [10, 25, 50, 100, 200, 300] # go upto a 100 , 'clf__criterion': ['gini', 'entropy', 'log_loss'] , 'clf__max_features': ['sqrt', 'log2', None] #deafult is sqrt , 'clf__min_samples_leaf': [1, 2, 3, 4, 5, 10]