diff options
Diffstat (limited to 'lib/caret.rb')
-rw-r--r-- | lib/caret.rb | 9 |
1 files changed, 8 insertions, 1 deletions
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." |