[ 
https://issues.apache.org/jira/browse/SPARK-6529?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14501444#comment-14501444
 ] 

Xusen Yin commented on SPARK-6529:
----------------------------------

[~josephkb] Word2Vec model is different with others. The Word2Vec model has two 
transforms, I think.

1. It transforms a Row(word: String) to a Row(code: Vector)
2. It also can transform the Row(word: String) to a Row(Array(similarityWords: 
String)), given a parameter call numSimilarityWords.

So I think we need make it a transformer from a Row(word: String) to Row(code: 
Vector, Array(similarityWords: String)), given the parameter numSimilarityWords.



> Word2Vec transformer
> --------------------
>
>                 Key: SPARK-6529
>                 URL: https://issues.apache.org/jira/browse/SPARK-6529
>             Project: Spark
>          Issue Type: Sub-task
>          Components: ML
>            Reporter: Xusen Yin
>            Assignee: Xusen Yin
>




--
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

Reply via email to