diff options
author | Christoph Helma <helma@in-silico.ch> | 2015-10-08 10:43:43 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2015-10-08 10:43:43 +0200 |
commit | 1a56148aadef031c4f487bc23fda43f4ac5b7369 (patch) | |
tree | 3555c5883ed0c292b105c40c185ebba3e5bd4e3e /test/prediction_models.rb | |
parent | 394d564699756288569169ff3e198d6d7702f092 (diff) | |
parent | e3217075b602a950a0ee995fcfa731d97b5ba3eb (diff) |
new master branch
Diffstat (limited to 'test/prediction_models.rb')
-rw-r--r-- | test/prediction_models.rb | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/test/prediction_models.rb b/test/prediction_models.rb new file mode 100644 index 0000000..1b9e788 --- /dev/null +++ b/test/prediction_models.rb @@ -0,0 +1,28 @@ +require_relative "setup.rb" + +class PredictionModelTest < MiniTest::Test + + def test_prediction_model + pm = Model::Prediction.from_csv_file "#{DATA_DIR}/hamster_carcinogenicity.csv" + #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.classification? + refute pm.regression? + pm.crossvalidations.each do |cv| + assert cv.accuracy > 0.75 + end + prediction = pm.predict Compound.from_smiles("CCCC(NN)C") + assert_equal "true", prediction[:value] + pm.delete + end +end |