Что эквивалентно «tree.value» DecisionTree из sklearn в Spark ML? - PullRequest
Новая
       36

Что эквивалентно «tree.value» DecisionTree из sklearn в Spark ML?

0 голосов
/ 28 февраля 2019

Сравнение моделей DecisionTree от Scikit-learn и Spark ML.Следующие очень близко соответствуют друг другу.Я не могу отобразить схему value

для Spark ML DecisionTree data: 

root
 |-- id: integer (nullable = true)
 |-- prediction: double (nullable = true)
 |-- impurity: double (nullable = true)
 |-- impurityStats: array (nullable = true)
 |    |-- element: double (containsNull = true)
 |-- gain: double (nullable = true)
 |-- leftChild: integer (nullable = true)
 |-- rightChild: integer (nullable = true)
 |-- split: struct (nullable = true)
 |    |-- featureIndex: integer (nullable = true)
 |    |-- leftCategoriesOrThreshold: array (nullable = true)
 |    |    |-- element: double (containsNull = true)
 |    |-- numCategories: integer (nullable = true)

sklearn tree.feature        --> sparkML root.split.featureIndex
sklearn tree.threshold      --> sparkML.root.leftCategoriesOrThreshold
sklearn tree.children_left  --> sparkML root.leftChild
sklearn tree.children_right --> sparkML root.rightChild
sklearn tree.value          --> sparkML root.?
...