summaryrefslogtreecommitdiff
path: root/test/prediction_models.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/prediction_models.rb')
-rw-r--r--test/prediction_models.rb20
1 files changed, 0 insertions, 20 deletions
diff --git a/test/prediction_models.rb b/test/prediction_models.rb
deleted file mode 100644
index a2e5fe2..0000000
--- a/test/prediction_models.rb
+++ /dev/null
@@ -1,20 +0,0 @@
-require_relative "setup.rb"
-
-class PredictionModelTest < MiniTest::Test
-
- def test_prediction_model
- pm = Model::Prediction.from_csv_file "#{DATA_DIR}/hamster_carcinogenicity.csv"
- [:endpoint,:species,:source].each do |p|
- refute_empty pm[p]
- end
- assert pm.classification?
- refute pm.regression?
- pm.crossvalidations.each do |cv|
- p 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 = pm.predict Compound.from_smiles("CCCC(NN)C")
- assert_equal "true", prediction[:value]
- pm.delete
- end
-end