Hao Ren created SPARK-10691:
-------------------------------
Summary: Make LogisticRegressionModel's evaluate method public
Key: SPARK-10691
URL: https://issues.apache.org/jira/browse/SPARK-10691
Project: Spark
Issue Type: Improvement
Components: ML
Affects Versions: 1.5.0
Reporter: Hao Ren
The following method in {{LogisticRegressionModel}} is marked as {{private}},
which prevents users from creating a summary on any given data set.
{code}
// TODO: decide on a good name before exposing to public API
private[classification] def evaluate(dataset: DataFrame)
: LogisticRegressionSummary = {
new BinaryLogisticRegressionSummary(
this.transform(dataset),
$(probabilityCol),
$(labelCol))
}
{code}
This method is definitely necessary to test model performance.
By the way, the name {{evaluate}} is already pretty good for me.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
