From 09452bba5c407c27721223d126e3f45c12b20a0c Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Thu, 13 Oct 2016 22:59:45 +0200 Subject: tests pass --- lib/caret.rb | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'lib/caret.rb') diff --git a/lib/caret.rb b/lib/caret.rb index df86093..2c4cd0c 100644 --- a/lib/caret.rb +++ b/lib/caret.rb @@ -9,6 +9,11 @@ module OpenTox if independent_variables.flatten.uniq == ["NA"] prediction = Algorithm::Regression::weighted_average dependent_variables:dependent_variables, weights:weights prediction[:warning] = "No variables for regression model. Using weighted average of similar substances." + elsif + dependent_variables.size < 3 + prediction = Algorithm::Regression::weighted_average dependent_variables:dependent_variables, weights:weights + prediction[:warning] = "Insufficient number of neighbors (#{dependent_variables.size}) for regression model. Using weighted average of similar substances." + else dependent_variables.each_with_index do |v,i| dependent_variables[i] = to_r(v) -- cgit v1.2.3