summaryrefslogtreecommitdiff
path: root/lib/regression.rb
blob: 3890987b9f2492eb42ff17a15ed756be9c507ae6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
module OpenTox
  module Algorithm
    
    class Regression

      def self.weighted_average dependent_variables:, independent_variables:nil, weights:, query_variables:nil
        # TODO: prediction_interval
        weighted_sum = 0.0
        sim_sum = 0.0
        dependent_variables.each_with_index do |v,i|
          weighted_sum += weights[i]*dependent_variables[i]
          sim_sum += weights[i]
        end if dependent_variables
        sim_sum == 0 ? prediction = nil : prediction = weighted_sum/sim_sum
        {:value => prediction}
      end

    end
  end
end