【发布时间】:2018-10-23 19:30:05
【问题描述】:
我是 ML 的新手,正在尝试对文本进行情绪检测。 所以我有一个 ISEAR 数据集,其中包含带有情感标签的推文。 所以我目前的准确率是 63%,我想提高到至少 70% 甚至更多。
代码如下:
inputs = Input(shape=(MAX_LENGTH, ))
embedding_layer = Embedding(vocab_size,
64,
input_length=MAX_LENGTH)(inputs)
# x = Flatten()(embedding_layer)
x = LSTM(32, input_shape=(32, 32))(embedding_layer)
x = Dense(10, activation='relu')(x)
predictions = Dense(num_class, activation='softmax')(x)
model = Model(inputs=[inputs], outputs=predictions)
model.compile(optimizer='adam',
loss='categorical_crossentropy',
metrics=['acc'])
model.summary()
filepath="weights-simple.hdf5"
checkpointer = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max')
history = model.fit([X_train], batch_size=64, y=to_categorical(y_train), verbose=1, validation_split=0.1,
shuffle=True, epochs=10, callbacks=[checkpointer])
【问题讨论】:
标签: python-3.x machine-learning keras deep-learning lstm