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

remove __estimator

Browse source
This commit is contained in:
Tanushree Tunstall 2022-05-22 23:39:35 +01:00
parent 1a1154b4f4
commit 9839b6f8d1
15 changed files with 79 additions and 79 deletions
Download patch file Download diff file Expand all files Collapse all files

16
uq_ml_models_FS/fs_UQ_RF.py
View file

@ -7,17 +7,17 @@ Created on Wed May 18 06:03:24 2022
"""
parameters = [
{
'clf__estimator': [RandomForestClassifier(**rs
'clf': [RandomForestClassifier(**rs
, **njobs
, bootstrap = True
, oob_score = True)],
'clf__estimator__max_depth': [4, 6, 8, 10, 12, 16, 20, None]
, 'clf__estimator__class_weight':['balanced','balanced_subsample']
, 'clf__estimator__n_estimators': [10, 25, 50, 100]
, 'clf__estimator__criterion': ['gini', 'entropy', 'log_loss']
, 'clf__estimator__max_features': ['sqrt', 'log2', None] #deafult is sqrt
, 'clf__estimator__min_samples_leaf': [1, 2, 3, 4, 5, 10]
, 'clf__estimator__min_samples_split': [2, 5, 15, 20]
'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__criterion': ['gini', 'entropy', 'log_loss']
, 'clf__max_features': ['sqrt', 'log2', None] #deafult is sqrt
, 'clf__min_samples_leaf': [1, 2, 3, 4, 5, 10]
, 'clf__min_samples_split': [2, 5, 15, 20]
}
]