From 8c6c59980bc82dc2177147f2fe34adf8bfbc1539 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Mon, 24 Aug 2015 12:17:07 +0200 Subject: Model::Prediction with tests, initial indexes --- test/prediction_models.rb | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 test/prediction_models.rb (limited to 'test/prediction_models.rb') diff --git a/test/prediction_models.rb b/test/prediction_models.rb new file mode 100644 index 0000000..b4ad415 --- /dev/null +++ b/test/prediction_models.rb @@ -0,0 +1,23 @@ +require_relative "setup.rb" + +class PredictionModelTest < MiniTest::Test + + def test_prediction_model + dataset = Dataset.from_csv_file "#{DATA_DIR}/hamster_carcinogenicity.csv" + model = Model::LazarFminerClassification.create dataset + cv = ClassificationCrossValidation.create model + metadata = JSON.parse(File.read("#{DATA_DIR}/hamster_carcinogenicity.json")) + + metadata[:model_id] = model.id + metadata[:crossvalidation_id] = cv.id + pm = Model::Prediction.new(metadata) + pm.save + [:endpoint,:species,:source].each do |p| + refute_empty pm[p] + end + assert pm.crossvalidation.accuracy > 0.8 + prediction = pm.predict Compound.from_smiles("CCCC(NN)C") + assert_equal "true", prediction[:value] + pm.delete + end +end -- cgit v1.2.3