From 8d2f1c8a0f6cc9f7a481d1117bf8b3351130b1ea Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Wed, 7 Oct 2015 12:34:02 +0200 Subject: generalised fingerprints --- test/lazar-regression.rb | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) (limited to 'test/lazar-regression.rb') diff --git a/test/lazar-regression.rb b/test/lazar-regression.rb index 8b2d473..4f5a332 100644 --- a/test/lazar-regression.rb +++ b/test/lazar-regression.rb @@ -4,23 +4,21 @@ class LazarRegressionTest < MiniTest::Test def test_weighted_average training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" - model = Model::LazarRegression.create training_dataset + model = Model::LazarRegression.create training_dataset, {:neighbor_algorithm_parameters => {:min_sim => 0}} compound = Compound.from_smiles "CC(C)(C)CN" prediction = model.predict compound - #p prediction - assert_equal 13.6, prediction[:value].round(1) - #assert_equal 0.83, prediction[:confidence].round(2) - assert_equal 1, prediction[:neighbors].size + assert_equal 7.2, prediction[:value].round(1) + assert_equal 91, prediction[:neighbors].size end def test_mpd_fingerprints training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" model = Model::LazarRegression.create training_dataset - model.neighbor_algorithm_parameters[:type] = "mpd" + model.neighbor_algorithm_parameters[:type] = "MP2D" compound = Compound.from_smiles "CCCSCCSCC" prediction = model.predict compound - assert_equal 0.04, prediction[:value].round(2) - assert_equal 1, prediction[:neighbors].size + assert_equal 0.02, prediction[:value].round(2) + assert_equal 3, prediction[:neighbors].size end def test_local_linear_regression -- cgit v1.2.3