diff options
author | Christoph Helma <helma@in-silico.ch> | 2015-09-14 13:44:51 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2015-09-14 13:44:51 +0200 |
commit | 33c2404882cbed78c1c5d9d0b60e0da4594a7fc5 (patch) | |
tree | d6c63300dc2258c231e6c5df3caf9c31d9c68ddd | |
parent | 4464b69ddf82fcb6302c8595217e00238517fe5f (diff) |
tests for Model::Prediction.[classification|regression]?
-rw-r--r-- | test/prediction_models.rb | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/test/prediction_models.rb b/test/prediction_models.rb index b4ad415..001ebcd 100644 --- a/test/prediction_models.rb +++ b/test/prediction_models.rb @@ -15,6 +15,8 @@ class PredictionModelTest < MiniTest::Test [:endpoint,:species,:source].each do |p| refute_empty pm[p] end + assert pm.classification? + refute pm.regression? assert pm.crossvalidation.accuracy > 0.8 prediction = pm.predict Compound.from_smiles("CCCC(NN)C") assert_equal "true", prediction[:value] |