tensorflow2 keras 搭建网络的两种方法

最新推荐文章于 2023-06-09 08:15:00 发布

李亮_sdu

最新推荐文章于 2023-06-09 08:15:00 发布

阅读量814

点赞数 1

分类专栏： Tebsorflow2 学习笔记文章标签： tensorflow

本文链接：https://blog.csdn.net/qq_40575024/article/details/105935412

版权

本文介绍了在TensorFlow2中利用Keras库搭建神经网络的两种方式：一是通过Sequential六步法，以鸢尾花数据集为例进行训练；二是采用Model继承，直接编写网络结构，同样展示了简单示例。

摘要由CSDN通过智能技术生成

tensorflow2 keras 搭建网络的两种方法

1. Sequential 六步法
2. Model 继承

1. Sequential 六步法

import    #import 相关模块
train test   #定义好测试集训练集
model=tf.keras.models.Sequential  # 定义网络结构
model.compile #优化器，损失函数
model.fit       # 训练过程
model.summary   # 训练过程中的参数

例如鸢尾花数据集的训练：

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

x_train = datasets.load_iris().data
y_train = datasets.load_iris().target

np.random.seed(116)
np.random.shuffle(x_train)
np.random.seed(116)
np.random.shuffle(y_train)
tf