Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-19054

Eliminate extra pass in NB

    XMLWordPrintableJSON

Details

    • Improvement
    • Status: Resolved
    • Minor
    • Resolution: Fixed
    • None
    • 2.2.0
    • ML
    • None

    Description

      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.

      Attachments

        Activity

          People

            podongfeng Ruifeng Zheng
            podongfeng Ruifeng Zheng
            Votes:
            0 Vote for this issue
            Watchers:
            2 Start watching this issue

            Dates

              Created:
              Updated:
              Resolved: