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

Apache Spark reassigned SPARK-18767:
------------------------------------

    Assignee: Apache Spark

> Unify Models' toString methods
> ------------------------------
>
>                 Key: SPARK-18767
>                 URL: https://issues.apache.org/jira/browse/SPARK-18767
>             Project: Spark
>          Issue Type: Improvement
>            Reporter: zhengruifeng
>            Assignee: Apache Spark
>            Priority: Minor
>
> Models' toString should output some info, no just the uid of its trainner. 
> {code}
> scala> val nb = new NaiveBayes
> nb: org.apache.spark.ml.classification.NaiveBayes = nb_18e8984091a8
> scala> val nbm = nb.fit(data)
> nbm: org.apache.spark.ml.classification.NaiveBayesModel = NaiveBayesModel 
> (uid=nb_18e8984091a8) with 2 classes
> scala> val dt = new DecisionTreeClassifier
> dt: org.apache.spark.ml.classification.DecisionTreeClassifier = 
> dtc_627dac64995e
> scala> val dtm = dt.fit(data)
> 16/12/07 15:08:14 WARN Executor: 1 block locks were not released by TID = 94:
> [rdd_8_0]
> dtm: org.apache.spark.ml.classification.DecisionTreeClassificationModel = 
> DecisionTreeClassificationModel (uid=dtc_627dac64995e) of depth 2 with 5 nodes
> scala> val lr = new LogisticRegression
> lr: org.apache.spark.ml.classification.LogisticRegression = 
> logreg_251625c948a0
> scala> val lrm = lr.fit(data)
> lrm: org.apache.spark.ml.classification.LogisticRegressionModel = 
> logreg_251625c948a0
> {code}
> I override toString in model to make them all like this:
> {{ModelClassName (uid=...) with key params}}



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

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

Reply via email to