Browse Source

Initial Word2Vec example.

tags/v0.9
Oceania2018 6 years ago
parent
commit
79eb262162
7 changed files with 44 additions and 0 deletions
  1. BIN
      data/text8.zip
  2. BIN
      graph/word2vec.meta
  3. +0
    -0
      test/TensorFlowNET.Examples/Text/DataHelpers.cs
  4. +0
    -0
      test/TensorFlowNET.Examples/Text/TextClassificationTrain.cs
  5. +0
    -0
      test/TensorFlowNET.Examples/Text/TextClassificationWithMovieReviews.cs
  6. +44
    -0
      test/TensorFlowNET.Examples/Text/Word2Vec.cs
  7. +0
    -0
      test/TensorFlowNET.Examples/Text/cnn_models/VdCnn.cs

BIN
data/text8.zip View File


BIN
graph/word2vec.meta View File


test/TensorFlowNET.Examples/TextClassification/DataHelpers.cs → test/TensorFlowNET.Examples/Text/DataHelpers.cs View File


test/TensorFlowNET.Examples/TextClassification/TextClassificationTrain.cs → test/TensorFlowNET.Examples/Text/TextClassificationTrain.cs View File


test/TensorFlowNET.Examples/TextClassification/TextClassificationWithMovieReviews.cs → test/TensorFlowNET.Examples/Text/TextClassificationWithMovieReviews.cs View File


+ 44
- 0
test/TensorFlowNET.Examples/Text/Word2Vec.cs View File

@@ -0,0 +1,44 @@
using System;
using System.Collections.Generic;
using System.Text;
using Tensorflow;

namespace TensorFlowNET.Examples
{
/// <summary>
/// Implement Word2Vec algorithm to compute vector representations of words.
/// https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/2_BasicModels/word2vec.py
/// </summary>
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 = "";
}
}
}

test/TensorFlowNET.Examples/TextClassification/cnn_models/VdCnn.cs → test/TensorFlowNET.Examples/Text/cnn_models/VdCnn.cs View File


Loading…
Cancel
Save