From fbded88db8b51f41ffbd5a02f601e4538ec87258 Mon Sep 17 00:00:00 2001 From: Christoph Helma Date: Fri, 14 Oct 2016 09:55:51 +0200 Subject: git commit added to model metadata --- lib/caret.rb | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'lib/caret.rb') diff --git a/lib/caret.rb b/lib/caret.rb index 2c4cd0c..e24c943 100644 --- a/lib/caret.rb +++ b/lib/caret.rb @@ -2,10 +2,17 @@ module OpenTox module Algorithm class Caret - # TODO classification # model list: https://topepo.github.io/caret/modelList.html def self.create_model_and_predict dependent_variables:, independent_variables:, weights:, method:, query_variables: + remove = [] + # remove independent_variables with single values + independent_variables.each_with_index { |values,i| remove << i if values.uniq.size == 1} + remove.sort.reverse.each do |i| + independent_variables.delete_at i + weights.delete_at i + query_variables.delete_at i + end 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." -- cgit v1.2.3