[jira] [Resolved] (SPARK-23071) spark ML print decision tree is not readable.

["Sean Owen (JIRA)" <ji...@apache.org>]

     [ https://issues.apache.org/jira/browse/SPARK-23071?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]

Sean Owen resolved SPARK-23071.
-------------------------------
    Resolution: Invalid

I'd start with these on the mailing list or StackOverflow please.

> spark ML print decision tree is not readable.
> ---------------------------------------------
>
>                 Key: SPARK-23071
>                 URL: https://issues.apache.org/jira/browse/SPARK-23071
>             Project: Spark
>          Issue Type: Improvement
>          Components: MLlib
>    Affects Versions: 2.2.1
>         Environment: Spark local[4], windows 7.
>            Reporter: David Hodeffi
>
> I have dataset which was created from dataframe using VectorAssembler, the dataframe is a transformation of StringIndexer() for multiple columns I trained my model :
> val assembler = new VectorAssembler().....
> val data = assembler.transform(...)
> val featureIndexer = new VectorIndexer()
> val gbt = new GBTRegressor()
>       .setLabelCol("label")
>       .setFeaturesCol("indexedFeatures")
>       .setMaxIter(10)
> val pipeline = new Pipeline()
>       .setStages(Array(featureIndexer, gbt))
> val model = pipeline.fit(trainingData)
> When I print my model it would look something like:
> GBTRegressionModel (uid=gbtr_24b22b08fa90) with 10 trees
>   Tree 0 (weight 1.0):
>     If (feature 16 <= 0.22222222222222224)
>      If (feature 16 <= 0.13333333333333333)
>       If (feature 16 <= 0.07142857142857144)
>        If (feature 16 <= 0.02222222222222222)....
> My first problem is that I would expect to see feature name and not feature index when printing the model, how can I resolve that? Another problem is caused because I used StringIndexer(), which means I would see the mapping of each value as int and not his string value. How can I print the model with the StringType column instead of the one was transformed with StringIndexer()?
> Thanks for helpers



--
This message was sent by Atlassian JIRA
(v6.4.14#64029)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org