search

rstudio / sparkxgb

R interface for XGBoost on Spark
https://spark.posit.co/packages/sparkxgb/
Other
47 stars 14 forks source link

XGBoostModel training failed for rsample::attrition in Spark 2.3 #18

Open javierluraschi opened 5 years ago

javierluraschi commented 5 years ago 
library(sparkxgb)
library(sparklyr)
library(dplyr)

sc <- spark_connect(master = "local", version = "2.3")

attrition <- copy_to(sc, rsample::attrition)
attrition

xgb_model <- xgboost_classifier(attrition,
                                Attrition ~ Age,
                                num_class = 2,
                                num_round = 50,
                                max_depth = 4)