Rthur2003 commited on
Commit
853f480
·
1 Parent(s): 94e94a1

feat: add additional imports for model evaluation and cross-validation

Browse files
Files changed (1) hide show
  1. app/training/generate_figures.py +2 -0
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]