Update proximity

BirkhoffG · Jun 13, 2023 · 174f2fa · 174f2fa
1 parent 34adcd3
commit 174f2fa
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/scripts/benchmark_carla.py b/scripts/benchmark_carla.py
@@ -19,11 +19,12 @@ def validity(factuals, counterfactuals, mlmodel):
 
 def proximity(factuals, counterfactuals):
     columns = counterfactuals.columns
-    return np.mean(
-        np.linalg.norm(factuals[columns].values - counterfactuals[columns].values, 
-        axis=1, ord=1)
+    prox = np.linalg.norm(
+            factuals[columns].values - counterfactuals[columns].values, 
+            axis=1, ord=1
     )
-
+    prox = prox[~np.isnan(prox)]
+    return np.mean(prox)
 
 def main():
     os.environ["CUDA_VISIBLE_DEVICES"]=""