ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 09:28:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

ObjectiveValueComponent.evaluate: remove rounding

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-15 08:26:31 -05:00
parent b1871869a0
commit 8d2d9873cb
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/python/miplearn/components/objective.py
View File

@@ -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),
},
}

4
src/python/miplearn/components/tests/test_objective.py
View File

@@ -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,
}
}