keras实战(二)--imdb影评分类/路透社语料多分类

Reference
N-gram模型
Keras深度神经网络训练IMDB情感分类的四种方法
Deep learning with Python

1.语料来源
由于下载太慢,可以使用以下链接下载,并将其拷贝到 ~/.keras/datasets目录下:
imdb链接:https://pan.baidu.com/s/1J9y40T3zIOlsMk0xHeVBqg 密码:y74j
reuters链接:https://pan.baidu.com/s/1R9kEK3-cCuU0YZpuXBqNDQ 密码:r7e4
2.代码:

from keras.datasets import reuters
import numpy as np
from keras.utils import np_utils
from keras.layers import Dense
from keras.models import Sequential
(train_data,train_labels), (test_data, test_labels) = reuters.load_data(num_words=10000)
# print(len(train_data))
# print(len(train_labels))
# print(train_labels[0])
def vectorize_sequences(sequences, dimension=10000):
    results = np.zeros((len(sequences), dimension))
    for i, sequence in enumerate(sequences):
        results[i, sequence] = 1.
    return results

x_train = vectorize_sequences(train_data)
x_test = vectorize_sequences(test_data)

y_train = np_utils.to_categorical(train_labels)
y_test = np_utils.to_categorical(test_labels)

model = Sequential()
model.add(Dense(64, activation='relu', input_shape=(10000,)))
model.add(Dense(64, activation='relu'))
model.add(Dense(46, activation='softmax'))
model.compile(optimizer='rmsprop',loss='categorical_crossentropy',metrics=['accuracy'])
x_val = x_train[:1000]
partial_x_train = x_train[1000:]
y_val = y_train[:1000]
partial_y_train = y_train[1000:]

model.fit(partial_x_train,
          partial_y_train,
          epochs=8,
          batch_size=512,
          validation_data=(x_val, y_val))
results = model.evaluate(x_test,y_test)
这里写代码片
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值