summaryrefslogtreecommitdiff
path: root/lib/model.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/model.rb')
-rw-r--r--lib/model.rb8
1 files changed, 4 insertions, 4 deletions
diff --git a/lib/model.rb b/lib/model.rb
index a0c60f0..966460b 100644
--- a/lib/model.rb
+++ b/lib/model.rb
@@ -481,13 +481,13 @@ module OpenTox
end
# TODO from_pubchem_aid
- def self.from_dataset training_dataset: , prediction_feature: , species: , endpoint:
+ def self.from_dataset training_dataset: , prediction_feature: , species: , endpoint: , folds: 10, repeats: 5
model_validation = Model::Validation.create species: species, endpoint: endpoint
- p "create model"
+ #p "create model"
model = Lazar.create training_dataset: training_dataset, prediction_feature: prediction_feature
model_validation[:model_id] = model.id
- p "create_crossvalidations"
- model_validation[:repeated_crossvalidation_id] = OpenTox::Validation::RepeatedCrossValidation.create(model).id # full class name required
+ #p "create_crossvalidations"
+ model_validation[:repeated_crossvalidation_id] = OpenTox::Validation::RepeatedCrossValidation.create(model,folds,repeats).id # full class name required
model_validation.save
model_validation
end