blob: cf6d9cb266390105323da6f559fa09e33036b7f0 (
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
# 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
|