blob: fd2855fcab03a32db57e8a1673f9089a9c1fbc64 (
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
26
|
module OpenTox
module Algorithm
# Regression algorithms
class Regression
# Weighted average
# @param [Array<TrueClass,FalseClass>] dependent_variables
# @param [Array<Float>] weights
# @return [Hash]
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
|