Skip to content

Commit 95efaec

Browse files
committed
Rename (xVariable, yVariable) -> (xs, ys)
1 parent dda0d79 commit 95efaec

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Linear Regression/LinearRegression.playground/Contents.swift

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,11 +37,11 @@ func multiply(_ a: [Double], _ b: [Double]) -> [Double] {
3737
return zip(a,b).map { $0.0 * $0.1 }
3838
}
3939

40-
func linearRegression(_ xVariable: [Double], _ yVariable: [Double]) -> ((Double) -> Double) {
41-
let sum1 = average(multiply(yVariable, xVariable)) - average(xVariable) * average(yVariable)
42-
let sum2 = average(multiply(xVariable, xVariable)) - pow(average(xVariable), 2)
40+
func linearRegression(_ xs: [Double], _ ys: [Double]) -> ((Double) -> Double) {
41+
let sum1 = average(multiply(ys, xs)) - average(xs) * average(ys)
42+
let sum2 = average(multiply(xs, xs)) - pow(average(xs), 2)
4343
let slope = sum1 / sum2
44-
let intercept = average(yVariable) - slope * average(xVariable)
44+
let intercept = average(ys) - slope * average(xs)
4545
return { intercept + slope * $0 }
4646
}
4747

0 commit comments

Comments
 (0)