Fix function used with Spark decision tree for regression

juliasilge · juliasilge · commit 1160e1e11170 · 2020-05-26T12:24:24.000-06:00
diff --git a/R/decision_tree_data.R b/R/decision_tree_data.R
@@ -260,7 +260,7 @@ set_fit(
     interface = "formula",
     data = c(formula = "formula", data = "x"),
     protect = c("x", "formula"),
-    func = c(pkg = "sparklyr", fun = "ml_decision_tree_classifier"),
+    func = c(pkg = "sparklyr", fun = "ml_decision_tree_regressor"),
     defaults =
       list(seed = expr(sample.int(10 ^ 5, 1)))
   )

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,7 @@ set_fit(`
`260`	`260`	`interface = "formula",`
`261`	`261`	`data = c(formula = "formula", data = "x"),`
`262`	`262`	`protect = c("x", "formula"),`
`263`		`- func = c(pkg = "sparklyr", fun = "ml_decision_tree_classifier"),`
	`263`	`+ func = c(pkg = "sparklyr", fun = "ml_decision_tree_regressor"),`
`264`	`264`	`defaults =`
`265`	`265`	`list(seed = expr(sample.int(10 ^ 5, 1)))`
`266`	`266`	`)`