Skip to content

Commit

Permalink
update default threshold
Browse files Browse the repository at this point in the history
  • Loading branch information
markus583 committed Mar 3, 2024
1 parent 153485a commit 0c0116d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion wtpsplit/train/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 0c0116d

Please sign in to comment.