diff options
author | Christoph Helma <helma@in-silico.ch> | 2011-04-05 19:05:59 +0200 |
---|---|---|
committer | Christoph Helma <helma@in-silico.ch> | 2011-04-05 19:05:59 +0200 |
commit | 4a3babce16c811ba3044c460533707b469b3c263 (patch) | |
tree | d5c690fadabc7375300f1a3eda92c8beecb223d3 /lazar.rb | |
parent | 16fdf49682898277566978a11df5e4d1ca7011c3 (diff) |
lazar model from ambit datasets
Diffstat (limited to 'lazar.rb')
-rw-r--r-- | lazar.rb | 117 |
1 files changed, 83 insertions, 34 deletions
@@ -8,62 +8,111 @@ class Float end end -class LazarRegressionTest < Test::Unit::TestCase +class LazarTest < Test::Unit::TestCase + + def setup + @predictions = [] + @models = [] + end + + def teardown + @predictions.each {|p| p.delete(@@subjectid)} + @models.each {|m| m.delete(@@subjectid)} + end =begin =end def test_create_regression_model model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => @@regression_training_dataset.uri, :subjectid => @@subjectid}).to_s - @lazar = OpenTox::Model::Lazar.find model_uri, @@subjectid - assert_equal 225, @lazar.features.size + lazar = OpenTox::Model::Lazar.find model_uri, @@subjectid + @models << lazar + assert_equal 1354, lazar.features.size compound = OpenTox::Compound.from_smiles("c1ccccc1NN") - prediction_uri = @lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid).to_s - @prediction = OpenTox::LazarPrediction.find(prediction_uri, @@subjectid) - assert_equal @prediction.value(compound).round_to(4), 0.1618.round_to(4) - assert_equal @prediction.confidence(compound).round_to(4), 0.6114.round_to(4) - assert_equal @prediction.neighbors(compound).size, 81 - @prediction.delete(@@subjectid) - @lazar.delete(@@subjectid) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid).to_s + prediction = OpenTox::LazarPrediction.find(prediction_uri, @@subjectid) + @predictions << prediction + assert_equal prediction.value(compound).round_to(4), 0.17.round_to(4) + assert_equal prediction.confidence(compound).round_to(4), 0.5697.round_to(4) + assert_equal prediction.neighbors(compound).size, 77 end -end -class LazarClassificationTest < Test::Unit::TestCase + def test_classification_model # create model model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => @@classification_training_dataset.uri, :subjectid => @@subjectid}).to_s - @lazar = OpenTox::Model::Lazar.find model_uri, @@subjectid - assert_equal @lazar.features.size, 41 + lazar = OpenTox::Model::Lazar.find model_uri, @@subjectid + @models << lazar + assert_equal lazar.features.size, 52 # single prediction compound = OpenTox::Compound.from_smiles("c1ccccc1NN") - prediction_uri = @lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) - @prediction = OpenTox::LazarPrediction.find(prediction_uri, @@subjectid) - assert_equal @prediction.value(compound), false - assert_equal @prediction.confidence(compound).round_to(4), 0.3005.round_to(4) - assert_equal @prediction.neighbors(compound).size, 15 - @prediction.delete(@@subjectid) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) + prediction = OpenTox::LazarPrediction.find(prediction_uri, @@subjectid) + @predictions << prediction + assert_equal prediction.value(compound), false + assert_equal prediction.confidence(compound).round_to(4), 0.3067.round_to(4) + assert_equal prediction.neighbors(compound).size, 14 # dataset activity compound = OpenTox::Compound.from_smiles("CNN") - prediction_uri = @lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) - @prediction = OpenTox::LazarPrediction.find prediction_uri, @@subjectid - assert !@prediction.measured_activities(compound).empty? - assert_equal @prediction.measured_activities(compound).first, true - assert @prediction.value(compound).nil? - @prediction.delete(@@subjectid) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) + prediction = OpenTox::LazarPrediction.find prediction_uri, @@subjectid + @predictions << prediction + assert !prediction.measured_activities(compound).empty? + assert_equal prediction.measured_activities(compound).first, true + assert prediction.value(compound).nil? # dataset prediction 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), @@subjectid - assert_equal @prediction.compounds.size, 4 - compound = OpenTox::Compound.new "http://ot-dev.in-silico.ch/compound/InChI=1S/C15H13NO/c1-10(17)16-13-6-7-15-12(9-13)8-11-4-2-3-5-14(11)15/h2-7,9H,8H2,1H3,(H,16,17)" - assert_equal @prediction.value(compound), nil - assert_equal @prediction.measured_activities(compound).first, true - @prediction.delete(@@subjectid) - @lazar.delete(@@subjectid) + prediction = OpenTox::LazarPrediction.find lazar.run(:dataset_uri => test_dataset.uri, :subjectid => @@subjectid), @@subjectid + @predictions << prediction + assert_equal prediction.compounds.size, 4 + compound = OpenTox::Compound.from_smiles "CC(=Nc1ccc2c(c1)Cc1ccccc21)O" + assert_equal prediction.value(compound), nil + assert_equal prediction.measured_activities(compound).first, true end + def test_ambit_classification_model + + # create model + dataset_uri = "http://apps.ideaconsult.net:8080/ambit2/dataset/9" + feature_uri ="http://apps.ideaconsult.net:8080/ambit2/feature/21573" + #model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => dataset_uri, :prediction_feature => feature_uri}).to_s + #lazar = OpenTox::Model::Lazar.find model_uri + model_uri = OpenTox::Algorithm::Lazar.new.run({:dataset_uri => dataset_uri, :prediction_feature => feature_uri, :subjectid => @@subjectid}).to_s + lazar = OpenTox::Model::Lazar.find model_uri, @@subjectid + assert_equal lazar.features.size, 6609 + #puts "Model: #{lazar.uri}" + #puts lazar.features.size + + # single prediction + compound = OpenTox::Compound.from_smiles("c1ccccc1NN") + #prediction_uri = lazar.run(:compound_uri => compound.uri) + #prediction = OpenTox::LazarPrediction.find(prediction_uri) + prediction_uri = lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) + prediction = OpenTox::LazarPrediction.find(prediction_uri, @@subjectid) + #puts "Prediction: #{prediction.uri}" + #puts prediction.value(compound) + assert_equal prediction.value(compound), true + #puts @prediction.confidence(compound).round_to(4) + #assert_equal @prediction.confidence(compound).round_to(4), 0.3005.round_to(4) + #assert_equal @prediction.neighbors(compound).size, 15 + #@prediction.delete(@@subjectid) + + # dataset activity + #compound = OpenTox::Compound.from_smiles("CNN") + #prediction_uri = @lazar.run(:compound_uri => compound.uri, :subjectid => @@subjectid) + #@prediction = OpenTox::LazarPrediction.find prediction_uri, @@subjectid + #assert !@prediction.measured_activities(compound).empty? + #assert_equal @prediction.measured_activities(compound).first, true + #assert @prediction.value(compound).nil? + #@prediction.delete(@@subjectid) + + # dataset prediction + #@lazar.delete(@@subjectid) + end + +end =begin =end -end |