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