diff options
author | gebele <gebele@in-silico.ch> | 2019-06-27 14:08:57 +0000 |
---|---|---|
committer | gebele <gebele@in-silico.ch> | 2019-06-27 14:08:57 +0000 |
commit | b536a45cf18b070cec3f9cb8a44fdac0bfa3c58e (patch) | |
tree | 76d560ea806369c2fae84fbe40815d2d92e4f429 /test/classification-validation.rb | |
parent | 22ff252ddcb8da087b50fc23ee22a857eb202738 (diff) |
fixed confidence value for cv stats; added tests
Diffstat (limited to 'test/classification-validation.rb')
-rw-r--r-- | test/classification-validation.rb | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/test/classification-validation.rb b/test/classification-validation.rb index 33f0353..5995c88 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 + end + end + when "predictivity", "true_rate" + av.each do |v| + refute_nil cv[type][key][v] + assert cv[type][key][v] > 0 + end + else + refute_nil cv[type][key] + assert cv[type][key] > 0 + end + end + end end end |