From 8519274487166d75b3b9ae28e61f7a7be9f7e83c Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Thu, 27 Oct 2016 11:58:07 +0200 Subject: probability plot for classification validations --- test/validation-classification.rb | 2 ++ 1 file changed, 2 insertions(+) (limited to 'test/validation-classification.rb') diff --git a/test/validation-classification.rb b/test/validation-classification.rb index b71e427..c93e71f 100644 --- a/test/validation-classification.rb +++ b/test/validation-classification.rb @@ -11,6 +11,8 @@ class ValidationClassificationTest < MiniTest::Test cv = ClassificationCrossValidation.create model assert cv.accuracy > 0.7, "Accuracy (#{cv.accuracy}) should be larger than 0.7, this may occur due to an unfavorable training/test set split" assert cv.weighted_accuracy > cv.accuracy, "Weighted accuracy (#{cv.weighted_accuracy}) should be larger than accuracy (#{cv.accuracy})." + #p cv + #File.open("tmp.pdf","w+"){|f| f.puts cv.probability_plot} end # parameters -- cgit v1.2.3