Tymec commited on
Commit
cc21abf
1 Parent(s): 419453c
Files changed (1) hide show
  1. 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.1, 1.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=7,
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__}")