diff --git a/data/text8.zip b/data/text8.zip new file mode 100644 index 00000000..436e05b2 Binary files /dev/null and b/data/text8.zip differ diff --git a/graph/word2vec.meta b/graph/word2vec.meta new file mode 100644 index 00000000..df120b7f Binary files /dev/null and b/graph/word2vec.meta differ diff --git a/test/TensorFlowNET.Examples/TextClassification/DataHelpers.cs b/test/TensorFlowNET.Examples/Text/DataHelpers.cs similarity index 100% rename from test/TensorFlowNET.Examples/TextClassification/DataHelpers.cs rename to test/TensorFlowNET.Examples/Text/DataHelpers.cs diff --git a/test/TensorFlowNET.Examples/TextClassification/TextClassificationTrain.cs b/test/TensorFlowNET.Examples/Text/TextClassificationTrain.cs similarity index 100% rename from test/TensorFlowNET.Examples/TextClassification/TextClassificationTrain.cs rename to test/TensorFlowNET.Examples/Text/TextClassificationTrain.cs diff --git a/test/TensorFlowNET.Examples/TextClassification/TextClassificationWithMovieReviews.cs b/test/TensorFlowNET.Examples/Text/TextClassificationWithMovieReviews.cs similarity index 100% rename from test/TensorFlowNET.Examples/TextClassification/TextClassificationWithMovieReviews.cs rename to test/TensorFlowNET.Examples/Text/TextClassificationWithMovieReviews.cs diff --git a/test/TensorFlowNET.Examples/Text/Word2Vec.cs b/test/TensorFlowNET.Examples/Text/Word2Vec.cs new file mode 100644 index 00000000..72dc0a80 --- /dev/null +++ b/test/TensorFlowNET.Examples/Text/Word2Vec.cs @@ -0,0 +1,44 @@ +using System; +using System.Collections.Generic; +using System.Text; +using Tensorflow; + +namespace TensorFlowNET.Examples +{ + /// + /// Implement Word2Vec algorithm to compute vector representations of words. + /// https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/2_BasicModels/word2vec.py + /// + public class Word2Vec : Python, IExample + { + public int Priority => 12; + public bool Enabled { get; set; } = true; + public string Name => "Word2Vec"; + + // Training Parameters + float learning_rate = 0.1f; + int batch_size = 128; + int num_steps = 3000000; + int display_step = 10000; + int eval_step = 200000; + + // Evaluation Parameters + string[] eval_words = new string[] { "five", "of", "going", "hardware", "american", "britain" }; + + public bool Run() + { + PrepareData(); + + var graph = tf.Graph().as_default(); + + tf.train.import_meta_graph("graph/word2vec.meta"); + + return false; + } + + public void PrepareData() + { + var url = ""; + } + } +} diff --git a/test/TensorFlowNET.Examples/TextClassification/cnn_models/VdCnn.cs b/test/TensorFlowNET.Examples/Text/cnn_models/VdCnn.cs similarity index 100% rename from test/TensorFlowNET.Examples/TextClassification/cnn_models/VdCnn.cs rename to test/TensorFlowNET.Examples/Text/cnn_models/VdCnn.cs