diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/model.rb | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/lib/model.rb b/lib/model.rb index 7cc6765..80affd5 100644 --- a/lib/model.rb +++ b/lib/model.rb @@ -68,10 +68,6 @@ module OpenTox :method => "fingerprint", :type => "MP2D", }, - :similarity => { - :method => "Algorithm::Similarity.tanimoto", - :min => 0.5, - }, :feature_selection => nil } @@ -79,10 +75,18 @@ module OpenTox model.algorithms[:prediction] = { :method => "Algorithm::Classification.weighted_majority_vote", } + model.algorithms[:similarity] = { + :method => "Algorithm::Similarity.tanimoto", + :min => 0.1, + } elsif model.class == LazarRegression model.algorithms[:prediction] = { :method => "Algorithm::Caret.rf", } + model.algorithms[:similarity] = { + :method => "Algorithm::Similarity.tanimoto", + :min => 0.5, + } end elsif substance_classes.first == "OpenTox::Nanoparticle" |