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

      def self.weighted_average descriptors:nil, neighbors:, parameters:nil, method:nil, relevant_features:nil
        # TODO: prediction_interval
        weighted_sum = 0.0
        sim_sum = 0.0
        neighbors.each do |neighbor|
          sim = neighbor["similarity"]
          activities = neighbor["measurements"]
          activities.each do |act|
            weighted_sum += sim*act
            sim_sum += sim
          end if activities
        end
        sim_sum == 0 ? prediction = nil : prediction = weighted_sum/sim_sum
        {:value => prediction}
      end

    end
  end
end