ruby_linear_regression 0.1.2 → 0.1.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/lib/ruby_linear_regression.rb +6 -4
- metadata +1 -1
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA1:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: 8cba2118afd13d70db9b3507b3cfee6a814b2e02
|
|
4
|
+
data.tar.gz: ba65dab424b629bb81f04811c1ddf4e24a910a85
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 9a4a4ebd99923983194a69c1cfeae6334ef0e6a7b9c5caa47c3f55ee92bddde8f8c30e735fd8b44bd3c38d4567ce4b6f3e0f1026277ebd1ecaac66b68df45b02
|
|
7
|
+
data.tar.gz: 02c69eb19ac21ce780767e83c1e239cb11e30aa3ac75837fff3404793e26ae429d28a8633147146f52ec47d83cad3a1a6bc3624ff02fd4116d8133195bbe5031
|
|
@@ -3,7 +3,7 @@ require 'matrix'
|
|
|
3
3
|
# RubyLinearRegression
|
|
4
4
|
class RubyLinearRegression
|
|
5
5
|
|
|
6
|
-
attr_reader :x,:y,:theta,:mu,:sigma
|
|
6
|
+
attr_reader :x,:y,:theta,:mu,:sigma, :lambda, :normalize
|
|
7
7
|
|
|
8
8
|
def initialize
|
|
9
9
|
@mu = 0
|
|
@@ -14,10 +14,12 @@ class RubyLinearRegression
|
|
|
14
14
|
# Arguments:
|
|
15
15
|
# x_data: (Two dimensiolnal array with the independent variables of your training data)
|
|
16
16
|
# y_data: (Array with the dependent variables of your training data)
|
|
17
|
-
def load_training_data x_data, y_data
|
|
17
|
+
def load_training_data x_data, y_data, normalize = true
|
|
18
|
+
|
|
19
|
+
@normalize = normalize
|
|
18
20
|
|
|
19
21
|
# normalize the x_data
|
|
20
|
-
x_data = normalize_data( x_data )
|
|
22
|
+
x_data = normalize_data( x_data ) if @normalize
|
|
21
23
|
|
|
22
24
|
# add 1 column to our data
|
|
23
25
|
x_data = x_data.map { |r| [1].concat(r) }
|
|
@@ -82,7 +84,7 @@ class RubyLinearRegression
|
|
|
82
84
|
# normalize
|
|
83
85
|
data.each_index do |i|
|
|
84
86
|
data[i] = (data[i] - @mu[i]) / @sigma[i].to_f
|
|
85
|
-
end
|
|
87
|
+
end if @normalize
|
|
86
88
|
|
|
87
89
|
# add 1 column to prediction data
|
|
88
90
|
data = [1].concat( data )
|