autoprognosis.utils.metrics module

evaluate_auc(y_test: ndarray, y_pred_proba: ndarray) → Tuple[float, float]

Helper for evaluating AUCROC/AUCPRC for any number of classes.

evaluate_brier_score(T_train: ndarray, Y_train: ndarray, Prediction: ndarray, T_test: ndarray, Y_test: ndarray, Time: float) → float

Helper for evaluating the Brier score.

evaluate_c_index(T_train: ndarray, Y_train: ndarray, Prediction: ndarray, T_test: ndarray, Y_test: ndarray, Time: float) → float

Helper for evaluating the C-INDEX metric.

generate_score(metric: ndarray) → Tuple[float, float]

get_y_pred_proba_hlpr(y_pred_proba: ndarray, nclasses: int) → ndarray

print_score(score: Tuple[float, float]) → str