modified loopity and multclass3 to have skf_cv as a parameters for cv

This commit is contained in:
Tanushree Tunstall 2022-03-17 18:17:58 +00:00
parent 97620c1bb0
commit d0c329a1d9
8 changed files with 161 additions and 127 deletions

View file

@ -138,6 +138,14 @@ parameters = [
#'tfidf__stop_words': [None],
'clf__estimator__alpha': (1e-2, 1e-3, 1e-1),
},
{
'clf__estimator': [LogisticRegression()],
'C': [0.001, 0.01, 0.1, 1, 10, 100, 1000],
'penalty': ['none', 'l1', 'l2', 'elasticnet'],
'max_iter': list(range(100,800,100)),
'solver': ['newton-cg', 'lbfgs', 'liblinear', 'sag', 'saga'],
},
]
pipeline = Pipeline([