From eef5d89a92dd7dde9acf9fc063a54e1fe729a89b Mon Sep 17 00:00:00 2001 From: gebele Date: Tue, 28 Mar 2017 11:00:12 +0000 Subject: fixed wrong accuracy assertion to rmse --- test/validation-regression.rb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/validation-regression.rb b/test/validation-regression.rb index 01ed644..c5ad312 100644 --- a/test/validation-regression.rb +++ b/test/validation-regression.rb @@ -84,7 +84,7 @@ class ValidationRegressionTest < MiniTest::Test repeated_cv = RepeatedCrossValidation.create model repeated_cv.crossvalidations.each do |cv| assert cv.r_squared > 0.34, "R^2 (#{cv.r_squared}) should be larger than 0.034" - assert_operator cv.accuracy, :>, 0.7, "model accuracy < 0.7, this may happen by chance due to an unfavorable training/test set split" + assert cv.rmse < 0.5, "RMSE (#{cv.rmse}) should be smaller than 0.5" end end -- cgit v1.2.3