tanu/ML_AI_training
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

n_estimators range to include upto a 1000 alex's suggestion

Browse source
This commit is contained in:
Tanushree Tunstall 2022-05-24 07:50:11 +01:00
parent b49c877f49
commit f2d2f361f0
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
uq_ml_models_FS/fs_UQ_RF.py
View file

@ -32,7 +32,7 @@ param_grid_rf = [
# 'clf': [RandomForestClassifier(**rs, **njobs, bootstrap = True, oob_score = True)], # 'clf': [RandomForestClassifier(**rs, **njobs, bootstrap = True, oob_score = True)],
'clf__max_depth': [4, 6, 8, 10, 12, 16, 20, None] 'clf__max_depth': [4, 6, 8, 10, 12, 16, 20, None]
, 'clf__class_weight':['balanced','balanced_subsample'] , '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__criterion': ['gini', 'entropy', 'log_loss']
, 'clf__max_features': ['sqrt', 'log2', None] #deafult is sqrt , 'clf__max_features': ['sqrt', 'log2', None] #deafult is sqrt
, 'clf__min_samples_leaf': [1, 2, 3, 4, 5, 10] , 'clf__min_samples_leaf': [1, 2, 3, 4, 5, 10]