Rthur2003 commited on
Commit
1d86705
·
1 Parent(s): 57f19bf

feat: define families dictionary for candidate classifiers in _build_candidate_families function

Browse files
Files changed (1) hide show
  1. app/training/train_classifier.py +1 -0
app/training/train_classifier.py CHANGED
@@ -370,6 +370,7 @@ def _class_ratio(y: np.ndarray) -> float:
370
 
371
 
372
  def _build_candidate_families(y: np.ndarray) -> dict[str, list[Any]]:
 
373
  "Logistic Regression": [
374
  LogisticRegression(
375
  C=value,
 
370
 
371
 
372
  def _build_candidate_families(y: np.ndarray) -> dict[str, list[Any]]:
373
+ families: dict[str, list[Any]] = {
374
  "Logistic Regression": [
375
  LogisticRegression(
376
  C=value,