Spaces:
Running
Running
Change df
Browse files- app/model.py +2 -2
app/model.py
CHANGED
@@ -36,7 +36,7 @@ def _identity(x: list[str]) -> list[str]:
|
|
36 |
def _get_vectorizer(
|
37 |
name: Literal["tfidf", "count", "hashing"],
|
38 |
n_features: int,
|
39 |
-
df: tuple[float, float] = (0
|
40 |
ngram: tuple[int, int] = (1, 2),
|
41 |
) -> TransformerMixin:
|
42 |
"""Get the appropriate vectorizer.
|
@@ -167,7 +167,7 @@ def train_model(
|
|
167 |
n_jobs=n_jobs,
|
168 |
# verbose=2,
|
169 |
scoring="accuracy",
|
170 |
-
n_iter=
|
171 |
)
|
172 |
|
173 |
pbar.set_description(f"Searching for {clf.__class__.__name__}")
|
|
|
36 |
def _get_vectorizer(
|
37 |
name: Literal["tfidf", "count", "hashing"],
|
38 |
n_features: int,
|
39 |
+
df: tuple[float, float] = (1.0, 1.0),
|
40 |
ngram: tuple[int, int] = (1, 2),
|
41 |
) -> TransformerMixin:
|
42 |
"""Get the appropriate vectorizer.
|
|
|
167 |
n_jobs=n_jobs,
|
168 |
# verbose=2,
|
169 |
scoring="accuracy",
|
170 |
+
n_iter=10,
|
171 |
)
|
172 |
|
173 |
pbar.set_description(f"Searching for {clf.__class__.__name__}")
|