Я использую RandomForestClassifier модель MLLIB в pyspark, и я хочу извлечь прогноз как вероятность (мне не нужен прогноз, такой как 0 и 1). Я нашел код, который позволяет мне извлечьвероятность предсказания ( код ), но теперь я не могу использовать эти функции для оценки модели:
MulticlassMetrics(labelsAndPredictions)
BinaryClassificationMetrics(labelsAndPredictions)
RegressionMetrics(labelsAndPredictions)
Есть ли у вас какие-либо предложения для «извлечения вероятности предсказания», а также использования этих функций?!