使用 Keras 分析 IMDB 电影数据 - 解决方案

4. 模型构建

使用 sequential 在这里构建模型。 请随意尝试不同的图层和大小! 此外,你可以尝试添加 dropout 以减少过拟合。


In [ ]:
# Building the model architecture with one layer of length 100
model = Sequential()
model.add(Dense(512, activation='relu', input_dim=1000))
model.add(Dropout(0.5))
model.add(Dense(num_classes, activation='softmax'))
model.summary()

# Compiling the model using categorical_crossentropy loss, and rmsprop optimizer.
model.compile(loss='categorical_crossentropy',
              optimizer='rmsprop',
              metrics=['accuracy'])

5. 训练模型

运行模型。 你可以尝试不同的 batch_size 和 epoch 数量!


In [ ]:
# Running and evaluating the model
hist = model.fit(x_train, y_train,
          batch_size=32,
          epochs=10,
          validation_data=(x_test, y_test), 
          verbose=2)