diff options
Diffstat (limited to 'test/validation.rb')
-rw-r--r-- | test/validation.rb | 16 |
1 files changed, 3 insertions, 13 deletions
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 |