summaryrefslogtreecommitdiff
path: root/test/regression-validation.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/regression-validation.rb')
-rw-r--r--test/regression-validation.rb17
1 files changed, 17 insertions, 0 deletions
diff --git a/test/regression-validation.rb b/test/regression-validation.rb
index 65bec63..9a2da8f 100644
--- a/test/regression-validation.rb
+++ b/test/regression-validation.rb
@@ -86,6 +86,23 @@ class RegressionValidationTest < MiniTest::Test
repeated_cv.crossvalidations.each do |cv|
assert cv.r_squared[:all] > 0.34, "R^2 (#{cv.r_squared[:all]}) should be larger than 0.34"
assert cv.rmse[:all] < 1.5, "RMSE (#{cv.rmse[:all]}) should be smaller than 0.5"
+=begin
+ #actually some values can be 0 or nil depending on the random folds in this small dataset
+ keys = cv.rmse.keys
+ types = ["rmse", \
+ "r_squared", \
+ "mae", \
+ "nr_predictions", \
+ "within_prediction_interval", \
+ "out_of_prediction_interval"
+ ]
+ types.each do |type|
+ keys.each do |key|
+ refute_nil cv[type][key]
+ assert cv[type][key] > 0
+ end
+ end
+=end
end
end