Train XGBoost with Spark

Train XGBoost with Spark

1
2
3
4
5
6
7
8
# XGB training script
# run spark-shell on cluster

spark-shell --name xxx --num-executors 15 --executor-cores 4 --executor-memory 20G --jars /tmp/xgboost4j-0.82.jar,/tmp/xgboost4j-spark-0.82.jar --driver-class-path /tmp/xgboost4j-0.82.jar,/tmp/xgboost4j-spark-0.82.jar

# import dependencies
import org.apache.spark.sql.types._
import scala.collection.mutable.ArrayBuilder

Comments

Your browser is out-of-date!

Update your browser to view this website correctly.&npsb;Update my browser now

×