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

14
uq_ml_models_FS/fs_UQ_DT.py
View file

@ -7,13 +7,13 @@ Created on Wed May 18 06:03:24 2022
"""
parameters = [
{
'clf__estimator': [DecisionTreeClassifier(**rs)]
, 'clf__estimator__max_depth': [None, 2, 4, 6, 8, 10, 12, 16, 20]
, 'clf__estimator__class_weight':['balanced']
, 'clf__estimator__criterion': ['gini', 'entropy', 'log_loss']
, 'clf__estimator__max_features': [None, 'sqrt', 'log2']
, 'clf__estimator__min_samples_leaf': [1, 2, 3, 4, 5, 10]
, 'clf__estimator__min_samples_split': [2, 5, 15, 20]
'clf': [DecisionTreeClassifier(**rs)]
, 'clf__max_depth': [None, 2, 4, 6, 8, 10, 12, 16, 20]
, 'clf__class_weight':['balanced']
, 'clf__criterion': ['gini', 'entropy', 'log_loss']
, 'clf__max_features': [None, 'sqrt', 'log2']
, 'clf__min_samples_leaf': [1, 2, 3, 4, 5, 10]
, 'clf__min_samples_split': [2, 5, 15, 20]
}
]