diff options
author | Christoph Helma <helma@in-silico.ch> | 2015-08-25 17:20:55 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2015-08-25 17:20:55 +0200 |
commit | f8faf510b4574df1a00fa61a9f0a1681fc2f4857 (patch) | |
tree | acdbe6666ca5f528be368c6f9fdf4d7fb51d031e /test/validation.rb | |
parent | 8c6c59980bc82dc2177147f2fe34adf8bfbc1539 (diff) |
Experiments added
Diffstat (limited to 'test/validation.rb')
-rw-r--r-- | test/validation.rb | 7 |
1 files changed, 5 insertions, 2 deletions
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 |