set default min sim to 0.1 for classification and 0.5 for regression

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"