diff options
Diffstat (limited to 'test/model-regression.rb')
-rw-r--r-- | test/model-regression.rb | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/model-regression.rb b/test/model-regression.rb index 644ca1c..86b927c 100644 --- a/test/model-regression.rb +++ b/test/model-regression.rb @@ -122,7 +122,6 @@ class LazarRegressionTest < MiniTest::Test assert_equal algorithms[:descriptors], model.algorithms[:descriptors] prediction = model.predict training_dataset.substances[10] refute_nil prediction[:value] - # TODO test predictin end def test_regression_with_feature_selection @@ -138,6 +137,8 @@ class LazarRegressionTest < MiniTest::Test assert_equal "Algorithm::Similarity.tanimoto", model.algorithms[:similarity][:method] assert_equal 0.1, model.algorithms[:similarity][:min] assert_equal algorithms[:feature_selection][:method], model.algorithms[:feature_selection][:method] + prediction = model.predict training_dataset.substances[10] + refute_nil prediction[:value] end def test_regression_parameters |