Python main Examples

Example #1

def trainWord2Vec(train_data, eval_data, save_path, vec_dim):
    FLAGS.train_data = train_data
    FLAGS.eval_data = eval_data
    FLAGS.save_path = save_path
    FLAGS.embedding_size = vec_dim

    word2vec.main([])
    print "done training data"

Example #2

File: word2vec_test.py Project: QPanProjects/Surrogate-Model

 def testWord2Vec(self):
     FLAGS.batch_size = 5
     FLAGS.num_neg_samples = 10
     FLAGS.epochs_to_train = 1
     FLAGS.min_count = 0
     word2vec.main([])

Example #3

File: word2vec_test.py Project: negar-rostamzadeh/tensorflow

 def testWord2Vec(self):
     FLAGS.batch_size = 5
     FLAGS.num_neg_samples = 10
     FLAGS.epochs_to_train = 1
     FLAGS.min_count = 0
     word2vec.main([])