probability plot for classification validations

1 files changed, 2 insertions, 0 deletions

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