diff options
author | Christoph Helma <helma@in-silico.ch> | 2016-05-07 10:34:03 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2016-05-07 10:34:03 +0200 |
commit | 48234554ea99b972a01718ac36c4e8332dd9159b (patch) | |
tree | 6f9d61a7b1c767a952063f69ed11449361395a05 /test/regression.rb | |
parent | 51f57e2858b60bed74ebcc97189b2188c900c283 (diff) |
-log10 for regression datasets, test cleanups
Diffstat (limited to 'test/regression.rb')
-rw-r--r-- | test/regression.rb | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/regression.rb b/test/regression.rb index 6d461ed..8ed8789 100644 --- a/test/regression.rb +++ b/test/regression.rb @@ -3,7 +3,7 @@ require_relative "setup.rb" class LazarRegressionTest < MiniTest::Test def test_weighted_average - training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" + training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi_log10.csv" model = Model::LazarRegression.create training_dataset.features.first, training_dataset, {:neighbor_algorithm_parameters => {:min_sim => 0}, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_weighted_average"} compound = Compound.from_smiles "CC(C)(C)CN" prediction = model.predict compound @@ -12,7 +12,7 @@ class LazarRegressionTest < MiniTest::Test end def test_mpd_fingerprints - training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" + training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi_log10.csv" model = Model::LazarRegression.create training_dataset.features.first, training_dataset model.neighbor_algorithm_parameters[:type] = "MP2D" compound = Compound.from_smiles "CCCSCCSCC" @@ -22,7 +22,7 @@ class LazarRegressionTest < MiniTest::Test end def test_local_fingerprint_regression - training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" + training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi_log10.csv" model = Model::LazarRegression.create(training_dataset.features.first, training_dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_fingerprint_regression") compound = Compound.from_smiles "NC(=O)OCCC" prediction = model.predict compound @@ -32,7 +32,7 @@ class LazarRegressionTest < MiniTest::Test end def test_local_physchem_regression - training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi.csv" + training_dataset = Dataset.from_csv_file "#{DATA_DIR}/EPAFHM.medi_log10.csv" model = Model::LazarRegression.create(training_dataset.features.first, training_dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_physchem_regression") compound = Compound.from_smiles "NC(=O)OCCC" prediction = model.predict compound |