-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path3-5-1.py
More file actions
103 lines (90 loc) · 3.47 KB
/
3-5-1.py
File metadata and controls
103 lines (90 loc) · 3.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#加载数据集
from tensorflow.python.keras.datasets import reuters
data_path = 'D:\\data\\reuters.npz'
(train_data,train_labels),(test_data,test_labels) = reuters.load_data(path=data_path,num_words=10000)
print(train_data.shape)
#将索引解码为新闻文本
word_index = reuters.get_word_index()
reverse_word_index = dict([(value,key)for (key,value) in word_index.items()])
decode_newswise = ' '.join([reverse_word_index.get(i-3,'?')for i in train_data[0]])
# print(decode_newswise)
#编码数据
import numpy as ny
def vectorize_sequences(sequences ,dimension = 10000):
results = ny.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)
#将标签数据向量化
def to_one_hot(labels,dimension = 46):
results = ny.zeros((len(labels),dimension))
for i ,label in enumerate(labels):
results[i,label] = 1.
return results
#将训练标签向量化
one_hot_train_labels = to_one_hot(train_labels)
#将测试标签向量化
one_hot_test_labels = to_one_hot(test_labels)
#下面使用keras中内置的方法来向量化训练标签和测试标签
from tensorflow.python.keras.utils.np_utils import to_categorical
one_hot_train_labels_1 = to_categorical(train_labels)
one_hot_test_labels_1 = to_categorical(test_labels)
# print("下面是自定义函数和keras内置方法对训练标签向量化和测试标签向量化的判断")
# print('训练标签向量化:',one_hot_train_labels == one_hot_train_labels_1)
# 以上结果:[[ True True True ... True True True]
# print('测试标签向量化:',one_hot_test_labels == one_hot_test_labels_1)
#以上结果:[[ True True True ... True True True]
#构建网络
from tensorflow.python.keras import layers,models
model = models.Sequential()
model.add(layers.Dense(64,activation='relu',input_shape=(10000,)))
model.add(layers.Dense(64,activation='relu'))
model.add(layers.Dense(46,activation='softmax'))
#编译模型
model.compile(
optimizer='rmsprop',
loss= 'categorical_crossentropy',
metrics=['acc']
)
#留出验证集
x_val = x_train[:1000]
partial_x_train = x_train[1000:]
y_val = one_hot_train_labels[:1000]
partial_y_train = one_hot_train_labels[1000:]
#训练模型
history = model.fit(
partial_x_train,
partial_y_train,
epochs=20,
batch_size=512,
validation_data=(x_val,y_val)
)
history_dict = history.history
#绘制训练损失和验证损失
import matplotlib.pyplot as plt
loss_values = history_dict['loss']
val_loss_values = history_dict['val_loss']
epochs = range(1,len(loss_values)+1)
plt.plot(epochs,loss_values,'bo',label='Training loss') #bo表示蓝色圆点
plt.plot(epochs,val_loss_values,'b',label='Validation loss') #b表示蓝色实线
plt.title('Training loss and Validation loss')
plt.xlabel('Epochs')
plt.ylabel('Loss')
plt.legend()
plt.show()
plt.clf() #清空图像
#绘制训练精度和验证精度
acc = history_dict['acc']
val_acc = history_dict['val_acc']
epochs = range(1,len(loss_values)+1)
plt.plot(epochs,acc,'bo',label='Training acc') #bo表示蓝色圆点
plt.plot(epochs,val_acc,'b',label='Validation acc') #b表示蓝色实线
plt.title('Training acc and Validation acc')
plt.xlabel('Epochs')
plt.ylabel('acc')
plt.legend()
plt.show()