Word2Vec Learner (Tensorflow)

To perform the actual training, hierarchical softmax and negative sampling are both available. The node uses Tensorflow as engine to speed up the pre-processing and to fit the model. Given the presence of a CUDA compatible NVIDIA GPU, training can be performed on the GPU.

Input ports

1. Type: Table
   Table with string column

   A KNIME table with a string column to use for Word2Vec training

Output ports

1. Type: Table
   Table of embeddings

   A KNIME table with three columns: the index of the token/the word, the token itself and the embedding for the token as a collection (KNIME native list).