tdc.evaluator

class tdc.evaluator.Evaluator(name)[source]

Bases: object

evaluator to evaluate predictions

Parameters

name (str) – the name of the evaluator function

assign_evaluator()[source]

obtain evaluator function given the evaluator name

tdc.evaluator.avg_auc(y_true, y_pred)[source]
tdc.evaluator.pcc(y_true, y_pred)[source]
tdc.evaluator.precision_at_recall_k(y_true, y_pred, threshold=0.9)[source]
tdc.evaluator.recall_at_precision_k(y_true, y_pred, threshold=0.9)[source]
tdc.evaluator.rmse(y_true, y_pred)[source]