summaryrefslogtreecommitdiff
path: root/lib/leave-one-out-validation.rb
diff options
context:
space:
mode:
authorhelma@in-silico.ch <helma@in-silico.ch>2018-11-02 20:34:44 +0100
committerhelma@in-silico.ch <helma@in-silico.ch>2018-11-02 20:34:44 +0100
commit3a9c9332b660d35720ad4fa1f55ee0883e53aecd (patch)
tree0c4b552ba434c4e03ea6e7808ead5a2d4ecb4cec /lib/leave-one-out-validation.rb
parent5b08a8c6d8e5567d253bec92d5bf5d18fd040cdc (diff)
warnings fixed, cleanup
Diffstat (limited to 'lib/leave-one-out-validation.rb')
-rw-r--r--lib/leave-one-out-validation.rb9
1 files changed, 1 insertions, 8 deletions
diff --git a/lib/leave-one-out-validation.rb b/lib/leave-one-out-validation.rb
index b0905b8..d37b6ce 100644
--- a/lib/leave-one-out-validation.rb
+++ b/lib/leave-one-out-validation.rb
@@ -16,18 +16,11 @@ module OpenTox
loo = klass.new :model_id => model.id
predictions = model.predict model.training_dataset.substances
predictions.each{|cid,p| p.delete(:neighbors)}
- nr_unpredicted = 0
predictions.each do |cid,prediction|
- if prediction[:value]
- prediction[:measurements] = model.training_dataset.values(cid, prediction[:prediction_feature_id])
- else
- nr_unpredicted += 1
- end
+ prediction[:measurements] = model.training_dataset.values(cid, prediction[:prediction_feature_id]) if prediction[:value]
predictions.delete(cid) unless prediction[:value] and prediction[:measurements]
end
predictions.select!{|cid,p| p[:value] and p[:measurements]}
- loo.nr_instances = predictions.size
- loo.nr_unpredicted = nr_unpredicted
loo.predictions = predictions
loo.statistics
$logger.debug "#{model.name}, LOO validation: #{Time.now-t} seconds"