TensorFlow2.0搭建神经网络最简单的入门

原创

于 2020-04-16 21:30:01 发布 · 2.5k 阅读

17 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #神经网络 #深度学习

本文根据北大TensorFlow2.0笔记，总结了搭建神经网络的六步法，并通过鸢尾花数据集实战，展示模型搭建过程。经过500次迭代，训练集损失为0.4425，准确率为0.8583；测试集损失为3130，准确率为0.9667。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

根据北大TensorFlow2.0笔记的内容，总结搭建神经网络的六步法
六部曲
以鸢尾花数据集实例来进行模型的搭建

import tensorflow as tf
from sklearn import datasets
import numpy as np

#导入训练集 以及训练集的标签
x_train = datasets.load_iris().data
y_train = datasets.load_iris().target
#实现对数据集的乱序
#seed( ) 用于指定随机数生成时所用算法开始的整数值。
#1.如果使用相同的seed( )值，则每次生成的随即数都相同；
np.random.seed(116