繁体   English   中英

带有Conv1D的Word2vec用于文本分类混乱

[英]Word2vec with Conv1D for text classification confusion

我正在做文本分类,并计划使用word2vec词嵌入并将其传递给Conv1D图层进行文本分类。 我有一个包含文本和相应标签(情感)的数据框 我使用了gensim模块,并使用了word2vec算法来生成词嵌入模型。 我使用的代码:

import pandas as pd
from gensim.models import Word2Vec
from nltk.tokenize import word_tokenize
df=pd.read_csv('emotion_merged_dataset.csv')
texts=df['text']
labels=df['sentiment']
df_tokenized=df.apply(lambda row: word_tokenize(row['text']), axis=1)
model = Word2Vec(df_tokenized, min_count=1)

我计划使用CNN并使用此词嵌入模型。 但是我应该如何为我的CNN使用这个词嵌入模型? 我应该输入什么?

我打算使用类似的东西(显然不使用相同的超参数):

model = Sequential()
model.add(layers.Embedding(max_features, 128, input_length=max_len))
model.add(layers.Conv1D(32, 7, activation='relu'))
model.add(layers.MaxPooling1D(5))
model.add(layers.Conv1D(32, 7, activation='relu'))
model.add(layers.GlobalMaxPooling1D())
model.add(layers.Dense(1))

有人可以帮我指出正确的方向吗? 提前致谢。

抱歉,您的回复很晚,希望它对您仍然有用。 根据您的应用程序,您可能需要下载特定的单词嵌入文件,例如,此处有手套文件

EMBEDDING_FILE='glove.6B.50d.txt'

embed_size = 50 # how big is each word vector
max_features = 20000 # how many unique words to use (i.e num rows in embedding vector)
maxlen = 100 # max number of words in a comment to use

word_index = tokenizer.word_index
nb_words = min(max_features, len(word_index))
embedding_matrix = np.random.normal(emb_mean, emb_std, (nb_words, embed_size))
for word, i in word_index.items():
    if i >= max_features: continue
    embedding_vector = embeddings_index.get(word)
    if embedding_vector is not None: embedding_matrix[i] = embedding_vector

#this is how you load the weights in the embedding layer
inp = Input(shape=(maxlen,))
x = Embedding(max_features, embed_size, weights=[embedding_matrix])(inp)

我从Jeremy Howard那里获取了这段代码,我想这就是您所需要的,如果您要加载其他文件,则过程非常相似,通常只需更改加载文件

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM