some metric changes

datamol-io · Aug 2, 2023 · 2d51bbb · 2d51bbb
1 parent ae71bdc
commit 2d51bbb
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/expts/neurips2023_configs/base_config/large.yaml b/expts/neurips2023_configs/base_config/large.yaml
@@ -364,6 +364,9 @@ metrics:
       metric: auroc_ipu
       num_classes: 5
       task: multiclass
+      target_to_int: True
+      target_nan_mask: -1000
+      ignore_index: -1000
       multitask_handling: mean-per-label
       threshold_kwargs: null
     - name: avpr
@@ -377,15 +380,18 @@ metrics:
       threshold_kwargs: null
   l1000_mcf7: *classif_metrics
   pcba_1328:
+  # use auroc and averageprecision (non_ipu version) so tha nans are handled correctly
     - name: auroc
-      metric: auroc_ipu
+      metric: auroc
       task: binary
       multitask_handling: mean-per-label
+      target_nan_mask: ignore
       threshold_kwargs: null
     - name: avpr
-      metric: average_precision_ipu
+      metric: averageprecision
       task: binary
       multitask_handling: mean-per-label
+      target_nan_mask: ignore
       threshold_kwargs: null
   pcqm4m_g25: &pcqm_metrics
     - name: mae