spark-user mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From AaronLee <...@wish.com.INVALID>
Subject Spark ml how to extract split points from trained decision tree mode
Date Thu, 11 Jun 2020 23:57:44 GMT
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-unsubscribe@spark.apache.org


Mime
View raw message