added feature selection on all models but lets see if it works, only worked until DT

This commit is contained in:
Tanushree Tunstall 2022-05-23 07:47:56 +01:00
parent a420822a93
commit 3c7d8690ee
17 changed files with 2425 additions and 1202 deletions

View file

@ -33,7 +33,7 @@ bts_jaccard 0.54
'bts_roc_auc': 0.65,
'bts_jaccard': 0.55}
#######################################################################
# RF: hyperparam [~45]
# RF: hyperparam [~45 min]
Best model:
{'clf__estimator': RandomForestClassifier(class_weight='balanced', max_depth=4, max_features=None,