diff options
Diffstat (limited to 'test/validation-regression.rb')
-rw-r--r-- | test/validation-regression.rb | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/test/validation-regression.rb b/test/validation-regression.rb index 7630521..01ed644 100644 --- a/test/validation-regression.rb +++ b/test/validation-regression.rb @@ -83,10 +83,9 @@ class ValidationRegressionTest < MiniTest::Test model = Model::Lazar.create training_dataset: dataset 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.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" end - File.open("tmp.png","w+"){|f| f.puts repeated_cv.correlation_plot} end end |