declaration

trainer.py

@@ -10,7 +10,7 @@ from sklearn.preprocessing import StandardScaler
 from sklearn.experimental import enable_iterative_imputer
 from sklearn.impute import KNNImputer, IterativeImputer
 
-from sklearn.metrics import accuracy_score, recall_score, f1_score, roc_auc_score, confusion_matrix
+from sklearn.metrics import accuracy_score, recall_score, f1_score, roc_auc_score, confusion_matrix , brier_score_loss
 from sklearn.utils.multiclass import type_of_target
 from sklearn.linear_model import LogisticRegression, Perceptron, SGDClassifier
 from sklearn.cross_decomposition import PLSRegression