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

Word2Vec uses a fixed length for sentences which is not reasonable for reality, and similarity functions and fields are not accessible

    XMLWordPrintableJSON

    Details

    • Type: Bug
    • Status: Resolved
    • Priority: Minor
    • Resolution: Fixed
    • Affects Version/s: 1.5.2
    • Fix Version/s: 2.0.0
    • Component/s: MLlib
    • Labels:
      None

      Description

      sentence boundary matters for sliding window, we shouldn't train model from a window across sentences.
      the current 1000 word as a hard split for sentences doesn't really make sense which is not consistent with both original c version or other implementation like deeplearning4j etc.
      the max sentence length is fixed and not tunable. Made it tunable as well.

      I made changes to address above issues.
      here is the pull request: https://github.com/apache/spark/pull/10152

        Attachments

          Activity

            People

            • Assignee:
              ygcao YongGang Cao
              Reporter:
              ygcao YongGang Cao
            • Votes:
              0 Vote for this issue
              Watchers:
              2 Start watching this issue

              Dates

              • Created:
                Updated:
                Resolved: