def test_mean_squared_error(caplog): """Unit test of mean_squared_error_scorer.""" caplog.set_level(logging.INFO) metric_dict = mean_squared_error_scorer(GOLDS, UNARY_PROBS, None) assert isequal(metric_dict, {"mean_squared_error": 0.1954166666666667}) metric_dict = mean_squared_error_scorer(PROB_GOLDS, PROBS, None) assert isequal(metric_dict, {"mean_squared_error": 0.16708333333333336})
def test_mean_squared_error(caplog): """Unit test of mean_squared_error_scorer""" caplog.set_level(logging.INFO) golds = np.array([3, -0.5, 2, 7]) probs = np.array([2.5, 0.0, 2, 8]) metric_dict = mean_squared_error_scorer(golds, probs, None) assert isequal(metric_dict, {"mean_squared_error": 0.375}) golds = np.array([[0.5, 1], [-1, 1], [7, -6]]) probs = np.array([[0, 2], [-1, 2], [8, -5]]) metric_dict = mean_squared_error_scorer(golds, probs, None) assert isequal(metric_dict, {"mean_squared_error": 0.7083333333333334})