summaryrefslogtreecommitdiff
path: root/test/classification-validation.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/classification-validation.rb')
-rw-r--r--test/classification-validation.rb28
1 files changed, 28 insertions, 0 deletions
diff --git a/test/classification-validation.rb b/test/classification-validation.rb
index 33f0353..b913e1e 100644
--- a/test/classification-validation.rb
+++ b/test/classification-validation.rb
@@ -36,6 +36,34 @@ class ClassificationValidationTest < MiniTest::Test
refute_nil validation.model.training_dataset_id
refute_equal model.training_dataset_id, validation.model.training_dataset_id
assert_equal params, validation_params
+ keys = cv.accuracy.keys
+ av = cv.accept_values
+ types = ["nr_predictions", \
+ "predictivity", \
+ "true_rate", \
+ "confusion_matrix"
+ ]
+ types.each do |type|
+ keys.each do |key|
+ case type
+ when "confusion_matrix"
+ cv[type][key].each do |arr|
+ arr.each do |a|
+ refute_nil a
+ assert a > 0, "#{cv[type][key]} values should be greater than 0."
+ end
+ end
+ when "predictivity", "true_rate"
+ av.each do |v|
+ refute_nil cv[type][key][v]
+ assert cv[type][key][v] > 0, "#{cv[type][key]} values should be greater than 0."
+ end
+ else
+ refute_nil cv[type][key]
+ assert cv[type][key] > 0, "#{cv[type][key]} value should be greater than 0."
+ end
+ end
+ end
end
end