Прогнозирование текста в python - PullRequest
0 голосов
/ 20 апреля 2020

enter image description here

У меня есть DataFrame, подобный этому на изображении. Я должен предсказать столбец "selected_text" на основе столбцов текста и настроений. Как я могу обучить модель на этих 2 столбцах для дальнейшего прогнозирования столбца selected_text?

Любая помощь будет признательна!

Заранее спасибо.

Ответы [ 2 ]

1 голос
/ 20 апреля 2020

Во-первых, столбец textID не имеет значения. Но, возможно, вы могли бы присвоить значение столбцу чувств (1 для положительного, 0 для отрицательного). Затем вы можете создать горячую кодировку для каждого слова в выбранном столбце, используя следующий код:

one_hot = []
current_bit = 1
current_one_hot_value = ""
for word in <EVERY_WORD_MENTIONED>:
    current_one_hot_value += bin(current_bit)[2:]
    for x in range(0,<HOW_MANY_WORDS> - len(current_one_hot_value)):
        current_one_hot_value += "0"
    one_hot.append(current_one_hot_value)
    current_one_hot_value = ""
    current_bit  = current_bit << 1

true_one_hot = []
one_hot_str = []
for encoding in one_hot:
    for bit in encoding:
        one_hot_str.append(int(bit))
    true_one_hot.append(one_hot_str)
    one_hot_str = []

Например, если бы у вас были слова "привет", "привет" и "пока", они бы стали: 001, 010 и 100.

Затем можно выполнить некоторую предварительную обработку фактической текстовой части фрейм данных и поместите его в нейронную сеть, такую ​​как на веб-сайте Keras (с большим количеством изменений):

from __future__ import print_function
from keras.callbacks import LambdaCallback
from keras.models import Sequential
from keras.layers import Dense
from keras.layers import LSTM
from keras.optimizers import RMSprop
from keras.utils.data_utils import get_file
import numpy as np
import random
import sys
import io

path = "<YOUR TEXT FILE OF TEXT HERE>"
with io.open(path, encoding='utf-8') as f:
    text = f.read().lower()
print('corpus length:', len(text))

chars = sorted(list(set(text)))
print('total chars:', len(chars))
char_indices = dict((c, i) for i, c in enumerate(chars))
indices_char = dict((i, c) for i, c in enumerate(chars))

# cut the text in semi-redundant sequences of maxlen characters
maxlen = 40
step = 3
sentences = []
next_chars = []
for i in range(0, len(text) - maxlen, step):
    sentences.append(text[i: i + maxlen])
    next_chars.append(text[i + maxlen])
print('nb sequences:', len(sentences))

print('Vectorization...')
x = np.zeros((len(sentences), maxlen, len(chars)), dtype=np.bool)
y = np.zeros((len(sentences), len(chars)), dtype=np.bool)
for i, sentence in enumerate(sentences):
    for t, char in enumerate(sentence):
        x[i, t, char_indices[char]] = 1
    y[i, char_indices[next_chars[i]]] = 1


# build the model: a single LSTM
print('Build model...')
model = Sequential()
model.add(LSTM(128, input_shape=(maxlen, len(chars))))
model.add(Dense(len(chars), activation='softmax'))

optimizer = RMSprop(learning_rate=0.01)
model.compile(loss='categorical_crossentropy', optimizer=optimizer)


def sample(preds, temperature=1.0):
    # helper function to sample an index from a probability array
    preds = np.asarray(preds).astype('float64')
    preds = np.log(preds) / temperature
    exp_preds = np.exp(preds)
    preds = exp_preds / np.sum(exp_preds)
    probas = np.random.multinomial(1, preds, 1)
    return np.argmax(probas)


def on_epoch_end(epoch, _):
    # Function invoked at end of each epoch. Prints generated text.
    print()
    print('----- Generating text after Epoch: %d' % epoch)

    start_index = random.randint(0, len(text) - maxlen - 1)
    for diversity in [0.2, 0.5, 1.0, 1.2]:
        print('----- diversity:', diversity)

        generated = ''
        sentence = text[start_index: start_index + maxlen]
        generated += sentence
        print('----- Generating with seed: "' + sentence + '"')
        sys.stdout.write(generated)

        for i in range(400):
            x_pred = np.zeros((1, maxlen, len(chars)))
            for t, char in enumerate(sentence):
                x_pred[0, t, char_indices[char]] = 1.

            preds = model.predict(x_pred, verbose=0)[0]
            next_index = sample(preds, diversity)
            next_char = indices_char[next_index]

            sentence = sentence[1:] + next_char

            sys.stdout.write(next_char)
            sys.stdout.flush()
        print()

print_callback = LambdaCallback(on_epoch_end=on_epoch_end)

model.fit(x, y,
          batch_size=128,
          epochs=60,
          callbacks=[print_callback])

Надеюсь, это то, что вы ищете, наилучшие пожелания:)

0 голосов
/ 20 апреля 2020

Вы упомянули, что знаете о классификации текста, но в этом случае вы хотите предсказать класс на основе двух входов вместо одного.

Если вы хотите предсказать текст (класс) из 2 входов, вы можете либо обучить две модели, каждая на каждом входе, а затем получить прогноз от среднего значения, или объединить эти два входа в один вход до обучения, а затем прогнозировать на основе этого одного входа

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...