From 0c0116d2d148a3cc82c33b64974568d404918427 Mon Sep 17 00:00:00 2001 From: markus583 Date: Sun, 3 Mar 2024 08:21:48 +0000 Subject: [PATCH] update default threshold --- wtpsplit/train/evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wtpsplit/train/evaluate.py b/wtpsplit/train/evaluate.py index c782f3ca..f3927165 100644 --- a/wtpsplit/train/evaluate.py +++ b/wtpsplit/train/evaluate.py @@ -37,7 +37,7 @@ def compute_f1(pred, true): ) -def get_metrics(labels, preds, threshold: float = 0.01): +def get_metrics(labels, preds, threshold: float = 0.5): # Compute precision-recall curve and AUC precision, recall, thresholds = sklearn.metrics.precision_recall_curve(labels, preds) pr_auc = sklearn.metrics.auc(recall, precision)