summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/regression-model.rb8
1 files changed, 8 insertions, 0 deletions
diff --git a/test/regression-model.rb b/test/regression-model.rb
index 3b41171..378ebc7 100644
--- a/test/regression-model.rb
+++ b/test/regression-model.rb
@@ -195,4 +195,12 @@ class LazarRegressionTest < MiniTest::Test
end
end
+ def test_fhm_prediction
+ #p File.join(File.dirname(__FILE__),"..","data","Acute_toxicity-Fathead_minnow.csv")
+ training_dataset = Dataset.from_csv_file File.join(File.dirname(__FILE__),"..","data","Acute_toxicity-Fathead_minnow.csv")
+ model = Model::Lazar.create training_dataset: training_dataset
+ prediction = model.predict Compound.from_smiles("N=Nc1ccccc1")
+ assert_equal 0.65, prediction[:neighbors][0][:measurement].round(2)
+ end
+
end