Я хотел бы знать, как получить вероятность каждой метки во время регрессионного задания, используя библиотеку PySpark ML.
У меня есть три вида этикеток, когда я использую GBTRegressor():
GBTR_model.transform(test_dataset)
Я получаю только вероятность относительно самого распространенного ярлыка, но я не получаю две другие вероятности относительно оставшихся ярлыков.