Skip to content

Commit f7ec3c4

Browse files
committed
Update linearRegression(_:_:) in README.md
1 parent 58a7f1c commit f7ec3c4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Linear Regression/README.markdown

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -123,11 +123,11 @@ We are using the ```map``` function to multiply each element.
123123
Finally, 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
```

0 commit comments

Comments
 (0)