200字范文,内容丰富有趣,生活中的好帮手!
200字范文 > Keras构建前馈神经网络并使用callbacks输出acc以及loss曲线(训练接 验证集)及效果可视化

Keras构建前馈神经网络并使用callbacks输出acc以及loss曲线(训练接 验证集)及效果可视化

时间:2020-12-19 13:43:46

相关推荐

Keras构建前馈神经网络并使用callbacks输出acc以及loss曲线(训练接 验证集)及效果可视化

Keras构建前馈神经网络并使用callbacks输出acc以及loss曲线(训练接、验证集)及效果可视化

在每个training/epoch/batch结束时,如果我们想执行某些任务,例如模型缓存、输出日志、计算当前的auc等等,Keras中的callback就派上用场了。

import numpy as npfrom keras.models import Sequentialfrom keras.layers import Dense,Dropoutimport tensorflow as tfimport keras# 如果使用GPU版TensorFlow,并且在显卡高占用率的情况下训练模型,# 那要在初始化Session时为其分配固定数量的显存,否则可能会在开始训练的时候直接报错退出def get_session():""" Construct a modified tf session."""config = tf.ConfigProto()config.gpu_options.allow_growth = Truereturn tf.Session(config=config)keras.backend.tensorflow_backend.set_session(get_session())#数据x_train = features_trainy_train = target_trainx_test = features_testy_test = target_test#第一层为输入层,20个节点#第二次为64个节点࿰

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。