
大数据文摘出品
来源:wandb
编译:邢畅、宁静
在训练神经网络的过程中,我们可能会希望可视化网络的性能和中间的结构,很多可视化代码的冗长复杂使得我们望而却步,有没有一行代码就能解决可视化的所有问题呢?
通过wandb,只需要加一行Python代码就可以可视化Keras网络性能指标和结构。(注:Keras使得构建神经网络变得简单明了,这一点深得人心)
这样好用的包如何下载呢?
只需运行“pip install wandb”,就可以轻松地安装wandb,然后所有的Keras示例就都可以运行了。
实测运行
实测运行
为了测试wandb的可视化效果,我修改了Keras examples目录中的一些脚本来测试这个功能,只需将下面一行代码添加到训练脚本的顶部。