zhengruifeng created SPARK-19054: ------------------------------------ Summary: Eliminate extra pass in NB Key: SPARK-19054 URL: https://issues.apache.org/jira/browse/SPARK-19054 Project: Spark Issue Type: Improvement Components: ML Reporter: zhengruifeng Priority: Minor
{{val numClasses = getNumClasses(dataset)}} in {{NaiveBayes}} are used to check the length of thresholds only when {{positiveLabel && isDefined(thresholds)}} is true. In the case that {{positiveLabel}} is true and {{isDefined(thresholds)}} is false, we don't need to calculate {{getNumClasses}} which may cause a pass on the whole data. -- 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