on pehe

pyg-team · May 12, 2024 · 4755bc4 · 4755bc4
1 parent a2b79d7
commit 4755bc4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/causalml.py b/examples/causalml.py
@@ -227,7 +227,7 @@ def eval(factual: TensorFrame, counterfactual: TensorFrame):
     within_rmse, within_rmse_cfact, within_error, within_pehe = eval(
         within_sample_tensor_frame, counterfactual_within_sample_tensor_frame)
 
-    if val_error < best_val_error:
+    if within_pehe < best_val_pehe:
         best_val_error = within_error
         best_test_error = test_error
         best_val_pehe = within_pehe