tensorflow2.0使用GradientTape自定义训练鸢尾花分类器
本文根据谷歌Tensorflow的官方事例进行介绍。官方事例在自定义训练:演示中查看。通过该实例可以更好的了解GradientTape的使用,尤其在自定义网络损失函数时尤其重要。 在原事例中使用的是url下载谷歌的鸢尾花数据集,由于网络原因下载失败,于是选用sklearn中的鸢尾花数据集。具体实现代码如下:import tensorflow as tfimport osimport ...
原创
2020-03-24 14:56:42 ·
385 阅读 ·
0 评论