You should be able to look at dtm.rootNode and, treating it as an InternalNode, get the .split from it
On Thu, Jun 11, 2020 at 7:02 PM AaronLee <y...@wish.com.invalid> wrote: > I am following official spark 2.4.3 tutorial > < > https://spark.apache.org/docs/2.4.3/ml-classification-regression.html#decision-tree-classifier> > > trained a decision tree model. How to extract split points from the trained > model? > > // 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 <- ? > > > > -- > Sent from: http://apache-spark-user-list.1001560.n3.nabble.com/ > > --------------------------------------------------------------------- > To unsubscribe e-mail: user-unsubscr...@spark.apache.org > >
