Как преобразовать текст в вектор, используя вложение word2ve c? - PullRequest
Новая
       71

Как преобразовать текст в вектор, используя вложение word2ve c?

0 голосов
/ 06 марта 2020

Предположим, у меня есть данные, показанные ниже:

| Текст

| Шторм в РИ хуже, чем в прошлом урагане

| Крушение зеленой линии в Чике go

| MEG выпускает Outlook в опасных условиях

Я создал модель word2ve c, используя следующий код: 

def sent_to_words(sentences):
    for sentence in sentences:
        yield(gensim.utils.simple_preprocess(str(sentence), deacc=True))

text_data = sent_to_words(df['Text'])
w2v_model = gensim.models.Word2Vec(text_data, size=100, min_count=1, window=5, iter=50)

Теперь, как я буду конвертировать текст, представленный в столбце «Текст» к векторам, использующим эту модель word2ve c?

1 Ответ

0 голосов
/ 06 марта 2020

вы можете получить сгенерированные вложения слов по 

w2v_model.wv

вы можете получить вложения слов определенного c слова по 

w2v_model.wv['word']
...