From f61b7d3c65d084747dc1bf87214e5ec0c57326be Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Tue, 9 Feb 2016 11:04:00 +0100 Subject: pls regression --- test/validation.rb | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) (limited to 'test/validation.rb') diff --git a/test/validation.rb b/test/validation.rb index 95f9bc0..066ec95 100644 --- a/test/validation.rb +++ b/test/validation.rb @@ -30,7 +30,7 @@ class ValidationTest < MiniTest::Test model = Model::LazarRegression.create dataset cv = RegressionCrossValidation.create model #cv = RegressionCrossValidation.find '561503262b72ed54fd000001' - #p cv.id + p cv #File.open("tmp.svg","w+"){|f| f.puts cv.correlation_plot} #`inkview tmp.svg` #File.open("tmp.svg","w+"){|f| f.puts cv.confidence_plot} @@ -71,6 +71,27 @@ class ValidationTest < MiniTest::Test assert cv.mae < 1 end + def test_pls_regression_crossvalidation + dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" + #dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.csv" + params = { + :prediction_algorithm => "OpenTox::Algorithm::Regression.local_pls_regression", + } + model = Model::LazarRegression.create dataset, params + cv = RegressionCrossValidation.create model + #p cv + cv.validation_ids.each do |vid| + model = Model::Lazar.find(Validation.find(vid).model_id) + p model + #assert_equal params[:neighbor_algorithm_parameters][:type], model[:neighbor_algorithm_parameters][:type] + #assert_equal params[:neighbor_algorithm_parameters][:min_sim], model[:neighbor_algorithm_parameters][:min_sim] + #refute_equal params[:neighbor_algorithm_parameters][:training_dataset_id], model[:neighbor_algorithm_parameters][:training_dataset_id] + end + + assert cv.rmse < 1.5, "RMSE > 1.5" + assert cv.mae < 1 + end + def test_repeated_crossvalidation dataset = Dataset.from_csv_file "#{DATA_DIR}/hamster_carcinogenicity.csv" model = Model::LazarClassification.create dataset -- cgit v1.2.3