summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lazar.rb10
1 files changed, 10 insertions, 0 deletions
diff --git a/lazar.rb b/lazar.rb
index b5f01e6..62a512b 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -80,6 +80,16 @@ class LazarTest < Test::Unit::TestCase
cleanup
end
+ def test_create_regression_prop_mlr_model
+ create_model :dataset_uri => @@regression_training_dataset.uri, :prediction_algorithm => "local_mlr_prop"
+ predict_compound OpenTox::Compound.from_smiles("c1ccccc1NN")
+ assert_equal 0.262.round_to(3), @predictions.first.confidence(@compounds.first).round_to(3)
+ assert_equal 123, @predictions.first.neighbors(@compounds.first).size
+ assert_equal 131, @model.features.size
+ assert_equal 0.374.round_to(3), @predictions.first.value(@compounds.first).round_to(3)
+ cleanup
+ end
+
def test_classification_model
create_model :dataset_uri => @@classification_training_dataset.uri
puts @model.uri