ml如何从训练好的决策树模型中提取分割点

bd1hkmkf 于 2021-05-29 发布在 Spark

关注(0)|答案(0)|浏览(185)

我遵循官方教程训练的决策树模型。如何从训练模型中提取分割点？

// model
val dt = new DecisionTreeClassifier()
  .setLabelCol("indexedLabel")
  .setFeaturesCol("indexedFeatures")
  .setMaxBins(10)

// Train model.  This also runs the indexers.
val dtm = dt.fit(trainingData)

// extract bin split points
how to do it                   <- ?

apache-spark apache-spark-ml

来源：https://stackoverflow.com/questions/62335281/spark-ml-how-to-extract-split-points-from-trained-decision-tree-model