文章の分類がしたいです。モデルを構築して、学習させることまではできました。次のコードにどんなコードを付け加えれば、この文章のカデゴリーはこれだ!と言う風に、予測が出来るようになりますか?
Xが文章(ベクトル)、yが分類するカテゴリです。
from keras.models import Sequential from keras.layers import Dense, Dropout from keras.layers import Embedding from keras.layers import LSTM embedding_dim = 50 model = Sequential() model.add(Embedding(input_dim=vocab_size, output_dim=embedding_dim, input_length=maxlen, trainable=True)) model.add(LSTM(128)) model.add(Dropout(0.5)) model.add(Dense(1, activation='sigmoid')) model.compile(loss='binary_crossentropy', optimizer='rmsprop', metrics=['accuracy']) history = model.fit(X_train, y_train, epochs=10, verbose=False, validation_data=(X_test, y_test), batch_size=10) from keras.backend import clear_session clear_session()
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。