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

最新推荐文章于 2025-03-11 11:14:18 发布

Healer__

最新推荐文章于 2025-03-11 11:14:18 发布

阅读量2.5k

点赞数

分类专栏： Tensorflow 文章标签： tensorflow 神经网络深度学习

本文链接：https://blog.youkuaiyun.com/Eugene__Li/article/details/105567199

版权

本文根据北大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