From e778475c578f13f30af4437845716d7e781c2609 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Sat, 13 Feb 2016 13:15:29 +0100 Subject: improved handling of duplicates in validations --- test/validation.rb | 16 +++------------- 1 file changed, 3 insertions(+), 13 deletions(-) (limited to 'test') diff --git a/test/validation.rb b/test/validation.rb index 066ec95..b1dc95e 100644 --- a/test/validation.rb +++ b/test/validation.rb @@ -73,21 +73,11 @@ class ValidationTest < MiniTest::Test 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", - } + 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 - + p cv.nr_instances + p cv.nr_unpredicted assert cv.rmse < 1.5, "RMSE > 1.5" assert cv.mae < 1 end -- cgit v1.2.3