AI coordinator blog

英文だけど、LSTMモデルを作成して文章を自動生成できるか試してみた

文章を自動生成できるようになれば、AIも大きく進化するなぁ〜なんて日々考えていますが、そう簡単にできるものではないですよね。

って自分に言い聞かせて、以前紹介した「wikipediaを学習した類義語を話すSlackbotの作成方法」記事以来、自然言語処理について半ば諦めていたところ、kerasのサンプルライブラリーに、LSTMモデルを活用した自動で文章を生成するライブラリーが公開されていることを知り、早速試してみたのでその内容の紹介になります。

英文ではありますが、文章の自動生成の可能性を感じることもできたので、興味がある方はチャレンジしてみてください。

参考ソースコード

以下のGitHubにありました。

https://github.com/fchollet/keras/blob/master/examples/lstm_text_generation.py

開発環境

iMac (27-inch, Late 2012)

プロセッサ 2.9 GHz intel Core i5

macOS Sierra バージョン 10.12.4

Anaconda3-4.2.0-MacOSX-x86_64

python 3.5.2

tensorflow 1.0.0

keras 1.2.2

ソースコード

ソースコードはLSTMモデルを保存するように改造したり、学習モデルの図式などで多少いじっています。

以下、学習モデルを作成するソースコードです。

100回学習するようにしています。

'''Example script to generate text from Nietzsche's writings.
At least 20 epochs are required before the generated text
starts sounding coherent.
It is recommended to run this script on GPU, as recurrent
networks are quite computationally intensive.
If you try this script on new data, make sure your corpus
has at least ~100k characters. ~1M is better.
'''

from __future__ import print_function
from keras.models import Sequential
from keras.layers import Dense, Activation
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

#学習結果の図式化用モジュール
from keras.utils.visualize_util import plot

#テキストの読み込み
path = get_file('nietzsche.txt', origin='https://s3.amazonaws.com/text-datasets/nietzsche.txt')
text = open(path).read().lower()
#コーパスの長さ
print('corpus length:', len(text))

chars = sorted(list(set(text)))
#文字数の表示
print('total chars:', len(chars))
#文字→ID
char_indices = dict((c, i) for i, c in enumerate(chars))
#ID→文字
indices_char = dict((i, c) for i, c in enumerate(chars))

# cut the text in semi-redundant sequences of maxlen characters

#テキストを40文字で区切る
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)))
model.add(Activation('softmax'))

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

#学習モデル図の作成
plot(model, to_file='LSTM_model.png')


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)

# train the model, output generated text after each iteration
#学習してテキストを生成する
for iteration in range(1, 100):
    print()
    print('-' * 50)
    print('Iteration', iteration)
    model.fit(X, y,
              batch_size=128,
              nb_epoch=1)
    
    #学習モデルの保存
    model.save('Keras_LSTM.h5');

    start_index = random.randint(0, len(text) - maxlen - 1)

    for diversity in [0.2, 0.5, 1.0, 1.2]:
        print()
        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 = np.zeros((1, maxlen, len(chars)))
            for t, char in enumerate(sentence):
                x[0, t, char_indices[char]] = 1.

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

            generated += next_char
            sentence = sentence[1:] + next_char

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

学習中は結果を以下のように次々と表示してくれます。

学習１７回目の状況はこんな感じでした。

--------------------------------------------------
Iteration 17
Epoch 1/1
200285/200285 [==============================] - 278s - loss: 1.3542   

----- diversity: 0.2
----- Generating with seed: "d the middle-class world and its yeas
an"
d the middle-class world and its yeas
and the subject, the subject, and the proservation of the spirit with an and desirality of the sperious and the same the subject, and the sense--and the spirit of the spirit of the spering to the subject, the sensible to the subject. the subject, the same spirit and the subject and the spirit of the spectare to the subject, the present the spirit and the spirit of the subject, and the sensible to th

----- diversity: 0.5
----- Generating with seed: "d the middle-class world and its yeas
an"
d the middle-class world and its yeas
and spirit and the still things are the imperience of the present can be no finer for the signific the propriness and the such an in the will of the same consider themselves to also and the morality and an and supersing to really when the sense--and it is an inflections of the and playerity and in the understood and morality, and the present have for the sight of the will to subtlence, and he will, 

----- diversity: 1.0
----- Generating with seed: "d the middle-class world and its yeas
an"
d the middle-class world and its yeas
anotherthing oy podic eusm of the dissing an anditionally best in -i have devils
pair) a sociounts themselves he
yetic of an extended, are erpons
to may redo no claimed to its band has also, the opposite comely, "ruliod," arouse to an impulse and wake earty
by should be authing all over can to able philosopherswhelloging the woild, the same gard and shadd will of all anysiud.

20. taetes and obey
vi

----- diversity: 1.2
----- Generating with seed: "d the middle-class world and its yeas
an"
d the middle-class world and its yeas
and dids blisdable
leadance to crut".--gamemens at the sen stoun dewli, admit of massnesss, thit. vanity.
falsearrism-but the superalin institutic prsfust. the mastored with as should rekneneufs, this opinianly dangerous
pifited, been man, inasmanbard  omay it
will, show that will, grets, in this anothaling lovece intons, crimounings, "and more calls has out give the in all
this worrwjult circlestly

次に保存した学習モデルに文字列を与えて、文章を生成するソースコードになります。

と言っても、上記のソースコードの必要な部分だけを切り抜いただけですけど・・・

diversity = 1.2　としていますので、この辺はお好みで。

与える文字列は40文字必要になります。

sentence = ‘how are you’にしています。

from keras.models import Sequential
from keras.layers import Dense, Activation, Dropout
from keras.layers import LSTM
from keras.optimizers import RMSprop
from keras.utils.data_utils import get_file
import numpy as np
import random, sys

from keras.models import load_model

model=load_model('Keras_LSTM.h5')

#テキストの読み込み
path = get_file('nietzsche.txt', origin='https://s3.amazonaws.com/text-datasets/nietzsche.txt')
text = open(path).read().lower()
chars = sorted(list(set(text)))
char_indices = dict((c, i) for i, c in enumerate(chars))
indices_char = dict((i, c) for i, c in enumerate(chars))

maxlen = 40

def sample(preds, temperature=1.0):
    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)

diversity = 1.2
generated = ''
#４０文字のテキスト作成する
sentence = 'how are you                             '
print(sentence)
sys.stdout.write(generated)
#与えたテキストを元に文を生成する
for i in range(50):
    x = np.zeros((1, maxlen, len(chars)))
    for t, char in enumerate(sentence):
        x[0, t, char_indices[char]] = 1.
    
    preds = model.predict(x, verbose=0)[0]
    next_index = sample(preds, diversity)
    next_char = indices_char[next_index]
    
    generated += next_char
    sentence = sentence[1:] + next_char
    sys.stdout.write(next_char)

実行結果

学習が１５回終了した時点での文章です。

how are you                             
 and
love constitntesations," and the
word more of

う〜ん。

意味不明ですね。

もっと学習回数を増やせば結果が変わって来るかもしれませんが、力尽きたので今日はこの辺で終了します。

それではまた。