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

saving work with scripts for feature selection

Browse source
This commit is contained in:
Tanushree Tunstall 2022-05-19 08:30:18 +01:00
parent a9dc3c43e5
commit fa0f5e5b39
3 changed files with 15 additions and 222 deletions
Download patch file Download diff file Expand all files Collapse all files

4
UQ_RF.py
View file

@ -39,8 +39,8 @@ parameters = [
'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': ['auto', 'sqrt']
, '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]
}