File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -123,11 +123,11 @@ We are using the ```map``` function to multiply each element.
123123Finally, the function which fits the line to the data:
124124
125125``` swift
126- func linearRegression (xVariable : [Double ], _ yVariable : [Double ]) -> (Double -> Double ) {
127- let sum1 = average (multiply (xVariable, yVariable )) - average (xVariable ) * average (yVariable )
128- let sum2 = average (multiply (xVariable, xVariable )) - pow (average (xVariable ), 2 )
126+ func linearRegression (xs : [Double ], _ ys : [Double ]) -> (Double -> Double ) {
127+ let sum1 = average (multiply (xs, ys )) - average (xs ) * average (ys )
128+ let sum2 = average (multiply (xs, xs )) - pow (average (xs ), 2 )
129129 let slope = sum1 / sum2
130- let intercept = average (yVariable ) - slope * average (xVariable )
130+ let intercept = average (ys ) - slope * average (xs )
131131 return { intercept + slope * $0 }
132132}
133133```
You can’t perform that action at this time.
0 commit comments