From 5e9a08c0b534fa96179fb5c81a9b4193e7b0aad8 Mon Sep 17 00:00:00 2001 From: "helma@in-silico.ch" Date: Mon, 29 Oct 2018 17:58:09 +0100 Subject: dataset folds fixed --- test/regression-model.rb | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'test/regression-model.rb') diff --git a/test/regression-model.rb b/test/regression-model.rb index 5903e88..0104741 100644 --- a/test/regression-model.rb +++ b/test/regression-model.rb @@ -168,4 +168,22 @@ class LazarRegressionTest < MiniTest::Test assert_equal 0.83, prediction[:value].round(2) end + def test_dataset_prediction + training_dataset = Dataset.from_csv_file File.join(DATA_DIR,"EPAFHM.medi_log10.csv") + model = Model::Lazar.create training_dataset: training_dataset + result = model.predict training_dataset + assert_kind_of Dataset, result + puts result.to_csv + puts result.features + # TODO + # check prediction + # check prediction_interval + # check warnings/applicability domain + assert 3, result.features.size + assert 8, result.compounds.size + assert_equal ["true"], result.values(result.compounds.first, result.features[1]) + assert_equal [0.65], result.values(result.compounds.first, result.features[2]) + assert_equal [0], result.values(result.compounds.first, result.features[2]) # classification returns nil, check if + end + end -- cgit v1.2.3