29 lines
787 B
Plaintext
29 lines
787 B
Plaintext
import "./matrix" for Matrix
|
|
|
|
var multipleRegression = Fn.new { |y, x|
|
|
var cy = y.transpose
|
|
var cx = x.transpose
|
|
return ((x * cx).inverse * x * cy).transpose[0]
|
|
}
|
|
|
|
var ys = [
|
|
Matrix.new([ [1, 2, 3, 4, 5] ]),
|
|
Matrix.new([ [3, 4, 5] ]),
|
|
Matrix.new([ [52.21, 53.12, 54.48, 55.84, 57.20, 58.57, 59.93, 61.29,
|
|
63.11, 64.47, 66.28, 68.10, 69.92, 72.19, 74.46] ])
|
|
]
|
|
|
|
var a = [1.47, 1.50, 1.52, 1.55, 1.57, 1.60, 1.63, 1.65, 1.68, 1.70, 1.73, 1.75, 1.78, 1.80, 1.83]
|
|
|
|
var xs = [
|
|
Matrix.new([ [2, 1, 3, 4, 5] ]),
|
|
Matrix.new([ [1, 2, 1], [1, 1, 2] ]),
|
|
Matrix.new([ List.filled(a.count, 1), a, a.map { |e| e * e }.toList ])
|
|
]
|
|
|
|
for (i in 0...ys.count) {
|
|
var v = multipleRegression.call(ys[i], xs[i])
|
|
System.print(v)
|
|
System.print()
|
|
}
|