Spaces:
Sleeping
Sleeping
feat: add additional imports for model evaluation and cross-validation
Browse files
app/training/generate_figures.py
CHANGED
|
@@ -32,6 +32,8 @@ from sklearn.metrics import (
|
|
| 32 |
precision_recall_curve, average_precision_score,
|
| 33 |
)
|
| 34 |
from sklearn.calibration import calibration_curve
|
|
|
|
|
|
|
| 35 |
|
| 36 |
# ── Paths ────────────────────────────────────────────────────────────────
|
| 37 |
BACKEND = Path(__file__).resolve().parents[2]
|
|
|
|
| 32 |
precision_recall_curve, average_precision_score,
|
| 33 |
)
|
| 34 |
from sklearn.calibration import calibration_curve
|
| 35 |
+
from sklearn.model_selection import StratifiedKFold, cross_val_predict
|
| 36 |
+
from sklearn.base import clone
|
| 37 |
|
| 38 |
# ── Paths ────────────────────────────────────────────────────────────────
|
| 39 |
BACKEND = Path(__file__).resolve().parents[2]
|