Merge branch 'joss_review_tiny_changes' of...

Merge branch 'joss_review_tiny_changes' of github.com:jrzaurin/pytorch-widedeep into joss_review_tiny_changes

tests/test_metrics/test_torchmetrics.py

@@ -77,6 +77,26 @@ def f2_score_multi(y_true, y_pred, average):
 @pytest.mark.parametrize(
     "metric_name, sklearn_metric, torch_metric",
     [
+        (
+            "Accuracy",
+            accuracy_score,
+            Accuracy(task="multiclass", num_classes=3, average="micro"),
+        ),
+        (
+            "Precision",
+            precision_score,
+            Precision(task="multiclass", num_classes=3, average="macro"),
+        ),
+        (
+            "Recall",
+            recall_score,
+            Recall(task="multiclass", num_classes=3, average="macro"),
+        ),
+        (
+            "F1Score",
+            f1_score,
+            F1Score(task="multiclass", num_classes=3, average="macro"),
+        ),
         (
             "Accuracy",
             accuracy_score,