mxnet.word2vec
2017-03-31 16:08:48 0 举报
AI智能生成
mxnet.word2vec是MXNet深度学习框架中用于实现Word2Vec模型的函数。Word2Vec是一种用于生成词向量的高效算法,它可以将文本中的每个单词映射到一个高维空间中的向量,使得语义上相似的单词在向量空间中距离较近。这种词向量表示可以用于自然语言处理任务,如文本分类、情感分析等。mxnet.word2vec函数提供了一种方便的方式来训练和加载Word2Vec模型,支持多种优化器和损失函数,以及自定义参数设置。通过使用mxnet.word2vec,用户可以在MXNet平台上轻松地实现自己的Word2Vec模型,从而提高自然语言处理任务的性能。
作者其他创作
大纲/内容
load_data
load file and split each line by space
calc term.freq, term.id and transfer dat by term.id
calc term.sample_array by pow(term.req, 0.75)
network_construct
Assume
1) data.shape: (#batch_size, #context_size);
2) label.shape: (#batch_size, #negative_sampling_size)
3) label_weight.shape: (#batch_size, #negative_sampling_size)
Steps
1. 将data embed 成(#batch_size, #context_size, #feature_size)
子主题
0 条评论
下一页