PySpark和MLLib和随机森林分类器：预测概率和模型评估

Question

我在pyspark中使用RandomForestClassifier的{​​{1}}模型，我想将预测提取为概率（不需要像0和1这样的预测） 我找到了可以提取预测概率（code）的代码 但是现在我无法使用这些功能进行模型评估：

MLLIB

您对“提取预测概率”以及使用这些功能有何建议？！