diff options
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 |