From 46c628f1757ce8274a0b277b3ec3306609b38c14 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Mon, 25 Jul 2016 15:53:22 +0200 Subject: local_weighted_average fallback fixed, cv predictions pulled from validations to avoid mongo document size errors --- lib/crossvalidation.rb | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'lib/crossvalidation.rb') diff --git a/lib/crossvalidation.rb b/lib/crossvalidation.rb index 7aae3d2..d7a1f08 100644 --- a/lib/crossvalidation.rb +++ b/lib/crossvalidation.rb @@ -18,7 +18,7 @@ module OpenTox cv.save # set created_at nr_instances = 0 nr_unpredicted = 0 - predictions = {} + #predictions = {} training_dataset = Dataset.find model.training_dataset_id training_dataset.folds(n).each_with_index do |fold,fold_nr| #fork do # parallel execution of validations can lead to Rserve and memory problems @@ -28,7 +28,7 @@ module OpenTox cv.validation_ids << validation.id cv.nr_instances += validation.nr_instances cv.nr_unpredicted += validation.nr_unpredicted - cv.predictions.merge! validation.predictions + #cv.predictions.merge! validation.predictions $logger.debug "Dataset #{training_dataset.name}, Fold #{fold_nr}: #{Time.now-t} seconds" #end end @@ -47,6 +47,12 @@ module OpenTox def validations validation_ids.collect{|vid| TrainTest.find vid} end + + def predictions + predictions = {} + validations.each{|v| predictions.merge!(v.predictions)} + predictions + end end class ClassificationCrossValidation < CrossValidation -- cgit v1.2.3