Spaces:
Sleeping
Sleeping
feat: define families dictionary for candidate classifiers in _build_candidate_families function
Browse files
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,
|