Skip to content

Commit 3c8481e

Browse files
committed
Spark *always* makes indicator variables, fix dependency for Spark + decision tree
1 parent 3a3c134 commit 3c8481e

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

R/decision_tree_data.R

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ set_pred(
232232

233233
set_model_engine("decision_tree", "classification", "spark")
234234
set_model_engine("decision_tree", "regression", "spark")
235-
set_dependency("decision_tree", "spark", "spark")
235+
set_dependency("decision_tree", "spark", "sparklyr")
236236

237237
set_model_arg(
238238
model = "decision_tree",
@@ -270,7 +270,7 @@ set_encoding(
270270
model = "decision_tree",
271271
eng = "spark",
272272
mode = "regression",
273-
options = list(predictor_indicators = FALSE)
273+
options = list(predictor_indicators = TRUE)
274274
)
275275

276276
set_fit(
@@ -291,7 +291,7 @@ set_encoding(
291291
model = "decision_tree",
292292
eng = "spark",
293293
mode = "classification",
294-
options = list(predictor_indicators = FALSE)
294+
options = list(predictor_indicators = TRUE)
295295
)
296296

297297
set_pred(

R/rand_forest_data.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -507,7 +507,7 @@ set_encoding(
507507
model = "rand_forest",
508508
eng = "spark",
509509
mode = "classification",
510-
options = list(predictor_indicators = FALSE)
510+
options = list(predictor_indicators = TRUE)
511511
)
512512

513513
set_fit(
@@ -527,7 +527,7 @@ set_encoding(
527527
model = "rand_forest",
528528
eng = "spark",
529529
mode = "regression",
530-
options = list(predictor_indicators = FALSE)
530+
options = list(predictor_indicators = TRUE)
531531
)
532532

533533
set_pred(

0 commit comments

Comments
 (0)