From de763211bd2b6451e3a8dc20eb95a3ecf72bef17 Mon Sep 17 00:00:00 2001 From: "helma@in-silico.ch" Date: Thu, 11 Oct 2018 12:13:40 +0200 Subject: initial dataset batch prediction --- test/model-classification.rb | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'test/model-classification.rb') diff --git a/test/model-classification.rb b/test/model-classification.rb index 7751bba..ca6eb27 100644 --- a/test/model-classification.rb +++ b/test/model-classification.rb @@ -94,6 +94,18 @@ class LazarClassificationTest < MiniTest::Test training_dataset.delete end + def test_dataset_prediction + training_dataset = Dataset.from_csv_file File.join(DATA_DIR,"hamster_carcinogenicity.csv") + model = Model::Lazar.create training_dataset: training_dataset + result = model.predict training_dataset + assert 3, result.features.size + assert 8, result.compounds.size + assert_equal ["true"], result.values(result.compounds.first, result.features[0]) + assert_equal [0.65], result.values(result.compounds.first, result.features[1]) + assert_equal [0], result.values(result.compounds.first, result.features[2]) # classification returns nil, check if + #p prediction_dataset + end + def test_carcinogenicity_rf_classification skip "Caret rf may run into a (endless?) loop for some compounds." dataset = Dataset.from_csv_file "#{DATA_DIR}/multi_cell_call.csv" -- cgit v1.2.3