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.
123
123
Finally, the function which fits the line to the data:
124
124
125
125
``` 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 )
129
129
let slope = sum1 / sum2
130
- let intercept = average (yVariable ) - slope * average (xVariable )
130
+ let intercept = average (ys ) - slope * average (xs )
131
131
return { intercept + slope * $0 }
132
132
}
133
133
```
You can’t perform that action at this time.
0 commit comments