| """Metrics.""" | |
| from sklearn.metrics import ( | |
| mean_squared_error, | |
| f1_score, | |
| precision_score, | |
| recall_score, | |
| roc_auc_score, | |
| accuracy_score, | |
| ) | |
| from src.utils.mapper import configmapper | |
| configmapper.map("metrics", "sklearn_f1")(f1_score) | |
| configmapper.map("metrics", "sklearn_p")(precision_score) | |
| configmapper.map("metrics", "sklearn_r")(recall_score) | |
| configmapper.map("metrics", "sklearn_roc")(roc_auc_score) | |
| configmapper.map("metrics", "sklearn_acc")(accuracy_score) | |
| configmapper.map("metrics", "sklearn_mse")(mean_squared_error) | |