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

10
uq_ml_models_FS/fs_UQ_XGB.py
View file

@ -18,11 +18,11 @@ Created on Wed May 18 06:03:24 2022
parameters = [
{
'clf__estimator': [XGBClassifier(**rs , **njobs, verbose = 3)]
, 'clf__estimator__learning_rate': [0.01, 0.05, 0.1, 0.2]
, 'clf__estimator__max_depth': [4, 6, 8, 10, 12, 16, 20]
#, 'clf__estimator__min_samples_leaf': [4, 8, 12, 16, 20]
#, 'clf__estimator__max_features': ['auto', 'sqrt']
'clf': [XGBClassifier(**rs , **njobs, verbose = 3)]
, 'clf__learning_rate': [0.01, 0.05, 0.1, 0.2]
, 'clf__max_depth': [4, 6, 8, 10, 12, 16, 20]
#, 'clf__min_samples_leaf': [4, 8, 12, 16, 20]
#, 'clf__max_features': ['auto', 'sqrt']
}
]