Skip to content

Commit

Permalink
fixed computation of orientability
Browse files Browse the repository at this point in the history
  • Loading branch information
danielbinschmid committed Aug 31, 2024
1 parent 159af67 commit 1836c48
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion code/metrics/accuracies.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ def compute_orientability_accuracies(
):
benchmarks = []
for metrics_ in metrics:
y_hat_ = torch.sigmoid(y_hat).long()
y_hat_ = torch.sigmoid(y_hat).round().long()
metric = metrics_.metric.to(y_hat.device)
benchmarks.append(
{"name": f"{name}_{metrics_.name}", "value": metric(y_hat_, y)}
Expand Down

0 comments on commit 1836c48

Please sign in to comment.