Unable to import ml.dmlc.xgboost4j.scala.spark.XGBoostClassifier

Community Discussions

Connect with fellow community members to discuss general topics related to the Databricks platform, industry trends, and best practices. Share experiences, ask questions, and foster collaboration within the community.

Hi I am trying to use the Scala XGBoostClassifier on Databricks, inspiring by this https://www.youtube.com/watch?v=vD3k-moYgmU video shared by Dbx official account. But I couldn't import the library.

I followed the official tutorial https://docs.databricks.com/en/machine-learning/train-model/xgboost-scala.html and installed the scala package.

However, when I imported this 'import ml.dmlc.xgboost4j.scala.spark.XGBoostClassifier' I got the error message 'ModuleNotFoundError: No module named 'ml.dmlc.xgboost4j.scala.spark.XGBoostClassifier''. Could you please shed light on how to correctly install this and import it? I would like to use the scala based XGBoost package to accelerate the computation. Thank you very much!

#xgboost #scala