ObjectiveValueComponent.evaluate: remove rounding

src/python/miplearn/components/objective.py

@@ -65,7 +65,7 @@ class ObjectiveValueComponent(Component):
                 "Explained variance": explained_variance_score(y_true_lb, y_pred_lb),
                 "Max error": max_error(y_true_lb, y_pred_lb),
                 "Mean absolute error": mean_absolute_error(y_true_lb, y_pred_lb),
-                "R2": np.round(r2_score(y_true_lb, y_pred_lb), 3),
+                "R2": r2_score(y_true_lb, y_pred_lb),
                 "Median absolute error": mean_absolute_error(y_true_lb, y_pred_lb),
             },
             "Upper bound": {
@@ -73,7 +73,7 @@ class ObjectiveValueComponent(Component):
                 "Explained variance": explained_variance_score(y_true_ub, y_pred_ub),
                 "Max error": max_error(y_true_ub, y_pred_ub),
                 "Mean absolute error": mean_absolute_error(y_true_ub, y_pred_ub),
-                "R2": np.round(r2_score(y_true_ub, y_pred_ub), 3),
+                "R2": r2_score(y_true_ub, y_pred_ub),
                 "Median absolute error": mean_absolute_error(y_true_ub, y_pred_ub),
             },
         }

src/python/miplearn/components/tests/test_objective.py

@@ -34,7 +34,7 @@ def test_obj_evaluate():
             'Mean absolute error': 126.5,
             'Mean squared error': 19194.5,
             'Median absolute error': 126.5,
-            'R2': -5.013,
+            'R2': -5.012843605607331,
         },
         'Upper bound': {
             'Explained variance': 0.0,
@@ -42,6 +42,6 @@ def test_obj_evaluate():
             'Mean absolute error': 126.5,
             'Mean squared error': 19194.5,
             'Median absolute error': 126.5,
-            'R2': -5.013,
+            'R2': -5.012843605607331,
         }
     }