File size: 551 Bytes
cc9c7ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
"""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)
|