saving work and wrapping up from the weekend

This commit is contained in:
Tanushree Tunstall 2022-05-23 00:31:02 +01:00
parent 9839b6f8d1
commit 1436557287
3 changed files with 12 additions and 13 deletions

View file

@ -66,7 +66,7 @@ print('\nbest model with feature selection:', fs_bmod)
pipe = Pipeline([
('pre', MinMaxScaler())
('selector', RFECV(LogisticRegression(**rs), cv = skf_cv, scoring = 'matthews_corrcoef'))
, ('selector', RFECV(LogisticRegression(**rs), cv = skf_cv, scoring = 'matthews_corrcoef'))
, ('classifier', LogisticRegression(**rs))])
search_space = [{'selector__min_features_to_select': [1,2]},