diff options
Diffstat (limited to 'test/model-validation.rb')
-rw-r--r-- | test/model-validation.rb | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/test/model-validation.rb b/test/model-validation.rb new file mode 100644 index 0000000..83986d6 --- /dev/null +++ b/test/model-validation.rb @@ -0,0 +1,19 @@ +require_relative "setup.rb" + +class ValidationModelTest < MiniTest::Test + + def test_validation_model + m = Model::Validation.from_csv_file "#{DATA_DIR}/hamster_carcinogenicity.csv" + [:endpoint,:species,:source].each do |p| + refute_empty m[p] + end + assert m.classification? + refute m.regression? + m.crossvalidations.each do |cv| + assert cv.accuracy > 0.74, "Crossvalidation accuracy (#{cv.accuracy}) should be larger than 0.75. This may happen due to an unfavorable training/test set split." + end + prediction = m.predict Compound.from_smiles("CCCC(NN)C") + assert_equal "true", prediction[:value] + m.delete + end +end |