zhengruifeng created SPARK-18767:
------------------------------------
Summary: Unify Models' toString methods
Key: SPARK-18767
URL: https://issues.apache.org/jira/browse/SPARK-18767
Project: Spark
Issue Type: Improvement
Reporter: zhengruifeng
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: [email protected]
For additional commands, e-mail: [email protected]
