Деревья решений Spark MLib: вероятность меток по функциям?

Я мог бы показать суммарные вероятности моегоlabelsНапример, после отображения моего дерева решений у меня есть таблица:

Total Predictions :
    65% impressions
    30% clicks
    5%  conversions

Но моя проблема заключается в том, чтобы найти вероятности (или рассчитать) поfeatures (по узлу), например:

if feature1 > 5
   if feature2 < 10
      Predict Impressions
      samples : 30 Impressions
   else feature2 >= 10
      Predict Clicks
      samples : 5 Clicks

Scikit делает это автоматически, я пытаюсь найти способ сделать это сSpark

Ответы на вопрос(1)

Ваш ответ на вопрос