shan_chandra
Databricks Employee
Databricks Employee

@amal15 - can you please include the below to the import statement and see if it works.

ml.dmlc.xgboost4j.scala.spark.XGBoostEstimator