diff options
author | mr <mr@mrautenberg.de> | 2010-12-16 10:51:36 +0100 |
---|---|---|
committer | mr <mr@mrautenberg.de> | 2010-12-16 10:51:36 +0100 |
commit | 1bc0f0d200cd25102d032dbce85e23c706e6ae72 (patch) | |
tree | 7aeb2f81c44cb2207b00ddb1607ddfc8b13a6519 /lazar.rb | |
parent | 5598135cbd2e9f4d0ac203f85069dea0468fda4d (diff) |
A&A for dataset and lazar
Diffstat (limited to 'lazar.rb')
-rw-r--r-- | lazar.rb | 27 |
1 files changed, 14 insertions, 13 deletions
@@ -7,48 +7,49 @@ class LazarTest < Test::Unit::TestCase =begin =end def test_create_regression_model - model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => @@regression_training_dataset.uri}).to_s + model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => @@regression_training_dataset.uri, :subjectid => @@subjectid}).to_s lazar = OpenTox::Model::Lazar.find model_uri assert_equal lazar.features.size, 222 compound = OpenTox::Compound.from_smiles("c1ccccc1NN") - prediction_uri = lazar.run(:compound_uri => compound.uri) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) prediction = OpenTox::LazarPrediction.find(prediction_uri) assert_equal prediction.value(compound), 0.149480050002539 assert_equal prediction.confidence(compound), 0.615246530364447 assert_equal prediction.neighbors(compound).size, 81 - prediction.delete - lazar.delete + prediction.delete(@@subjectid) + lazar.delete(@@subjectid) end def test_default_classification_model # create model - model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => @@classification_training_dataset.uri}).to_s + model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => @@classification_training_dataset.uri, :subjectid => @@subjectid}).to_s lazar = OpenTox::Model::Lazar.find model_uri assert_equal lazar.features.size, 41 # single prediction compound = OpenTox::Compound.from_smiles("c1ccccc1NN") - prediction_uri = lazar.run(:compound_uri => compound.uri) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) prediction = OpenTox::LazarPrediction.find(prediction_uri) assert_equal prediction.value(compound), false assert_equal prediction.confidence(compound), 0.25857114104619 assert_equal prediction.neighbors(compound).size, 15 - prediction.delete + prediction.delete(@@subjectid) # dataset activity compound = OpenTox::Compound.from_smiles("CNN") - prediction = OpenTox::LazarPrediction.find lazar.run(:compound_uri => compound.uri) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) + prediction = OpenTox::LazarPrediction.find prediction_uri assert !prediction.measured_activities(compound).empty? puts prediction.measured_activities(compound).first.inspect assert_equal prediction.measured_activities(compound).first, true assert prediction.value(compound).nil? - prediction.delete + prediction.delete(@@subjectid) # dataset prediction - test_dataset = OpenTox::Dataset.create_from_csv_file("data/multicolumn.csv") - prediction = OpenTox::LazarPrediction.find lazar.run(:dataset_uri => test_dataset.uri) + test_dataset = OpenTox::Dataset.create_from_csv_file("data/multicolumn.csv", @@subjectid) + prediction = OpenTox::LazarPrediction.find lazar.run(:dataset_uri => test_dataset.uri, :subjectid => @@subjectid) assert_equal prediction.compounds.size, 4 compound = OpenTox::Compound.new prediction.compounds.first assert_equal prediction.value(compound), false - prediction.delete - lazar.delete + prediction.delete(@@subjectid) + lazar.delete(@@subjectid) end end |