我正在使用Pyspark并遵循本教程，但存在多类分类问题。 https://databricks-prod-cloudfront.cloud.databricks.com/public/4027ec902e239c93eaaa8714f173bcfc/88328598478746/1288525732792552/576313034656021/latest.html
预测输出（实际上是预测或detail_predictions）没有将字符串标签清晰地映射到概率标签或索引。
我如何知道LabelA，LabelB等映射到什么位置以找到具有最大概率的标签？
我尝试查看xgboost类可用的方法，并将detail_predictions设置为yes。
model = estimator.fit(train)
predicted = model.transform(for_predictions)
display(predicted)