From f8faf510b4574df1a00fa61a9f0a1681fc2f4857 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Tue, 25 Aug 2015 17:20:55 +0200 Subject: Experiments added --- test/validation.rb | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'test/validation.rb') diff --git a/test/validation.rb b/test/validation.rb index 485769c..009c337 100644 --- a/test/validation.rb +++ b/test/validation.rb @@ -24,8 +24,8 @@ class ValidationTest < MiniTest::Test end def test_regression_crossvalidation - dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" - #dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.csv" + #dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" + dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.csv" model = Model::LazarRegression.create dataset cv = RegressionCrossValidation.create model p cv.rmse @@ -33,6 +33,9 @@ class ValidationTest < MiniTest::Test p cv.mae p cv.weighted_mae #`inkview #{cv.plot}` + #puts JSON.pretty_generate(cv.misclassifications)#.collect{|l| l.join ", "}.join "\n" + p cv.misclassifications.collect{|l| l[:neighbors].size} + `inkview #{cv.plot}` assert cv.rmse < 30, "RMSE > 30" assert cv.weighted_rmse < cv.rmse, "Weighted RMSE (#{cv.weighted_rmse}) larger than unweighted RMSE(#{cv.rmse}) " assert cv.mae < 12 -- cgit v1.2.3