栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

《Python 深度学习》7.2.2 TensorBoard简介(代码)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

《Python 深度学习》7.2.2 TensorBoard简介(代码)

1. 使用了TensorBoard的文本分类模型
import keras
from keras import layers
from keras.datasets import imdb
from keras.preprocessing import sequence

max_features = 2000 #作为特征的单词个数
max_len = 500 #在这么多单词之后截断文本

(x_train, y_train), (x_test, y_test) = imdb.load_data(num_words=max_features)
x_train = sequence.pad_sequences(x_train, maxlen=max_len)
print(x_train.shape)    #(25000, 500)
x_test = sequence.pad_sequences(x_test, maxlen=max_len)
print(x_test.shape) 

(25000, 500)

(25000, 500)

model = keras.models.Sequential()
model.add(layers.Embedding(max_features, 128, input_length=max_len, name='embed'))
model.add(layers.Conv1D(32, 7, activation='relu'))
model.add(layers.MaxPooling1D(5))
model.add(layers.Conv1D(32, 7, activation='relu'))
model.add(layers.GlobalMaxPooling1D())
model.add(layers.Dense(1))
model.summary()
#由于电脑内存过小,无法写入25000行的数据,因此取前1000行
print(x_train[:1000].shape)   #(1000, 500)

model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics=['acc'])

 2. 在使用TensorBoard之前,首先需要创建一个目录,用来保存生成的日志文件,可手动创建也可使用终端命令创建。

$ mkdir my_log_dir

3. 使用一个TensorBoard回调函数来训练模型
callbacks = [
    keras.callbacks.TensorBoard(
        log_dir='my_log_dir',  #将日志写入这个位置
        histogram_freq=1,  #每一轮之后记录激活的直方图
        embeddings_freq=1,  #每一轮之后记录嵌入数据
        embeddings_data=x_train[:1000].astype('float32')  #指定embeddings_data的值
    )
]
history = model.fit(x_train, y_train,
                    epochs=20,
                    batch_size=128,
                    validation_split=0.2,
                    callbacks=callbacks
                   )

4. 在命令行启动TensorBoard服务器,指示它读取回调函数当前正在写入的日志

tensorboard --logdir "D:2022ThesisDeep Learning with PythonCodemy_log_dir"

 5. 利用keras.utils.plot_model函数,将模型绘制为层组成的图 
from keras.utils import plot_model
plot_model(model,to_file='model.png')

 报错还未解决!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/581020.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号