【tensorflow框架神经网络实现鸢尾花分类_Keras】

文章目录

1、前言

【tensorflow框架神经网络实现鸢尾花分类】一文中使用自定义的方式,实现了鸢尾花数据集的分类工作。在这里使用tensorflow中的keras模块快速、极简实现鸢尾花分类任务。

2、鸢尾花分类

python 复制代码
import tensorflow as tf
from sklearn import datasets
import numpy as np

# 加载数据集
np.random.seed(0)
iris = datasets.load_iris()
x_train, y_train = iris.data, iris.target
np.random.seed(0)
np.random.shuffle(x_train)
np.random.seed(0)
np.random.shuffle(y_train)

# 设置随机种子
tf.random.set_seed(0)

# 构建模型
model = tf.keras.models.Sequential([
    tf.keras.layers.Dense(3, activation='softmax', kernel_regularizer=tf.keras.regularizers.l2(0.01))
])

# 编译模型
model.compile(optimizer=tf.keras.optimizers.SGD(learning_rate=0.1),
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=False),
              metrics=['sparse_categorical_accuracy'])

# 训练模型
model.fit(x_train, y_train, batch_size=32, epochs=500, validation_split=0.2, validation_freq=20)

# 打印模型摘要
model.summary()

3、结果打印

相关推荐
心动啊12115 小时前
深度神经网络1——梯度问题+标签数不够问题
人工智能·神经网络·dnn
ViperL116 小时前
[智能算法]可微的神经网络搜索算法-FBNet
人工智能·深度学习·神经网络
HenrySmale19 小时前
05 回归问题和分类问题
分类·数据挖掘·回归
✎﹏赤子·墨筱晗♪19 小时前
深入浅出LVS负载均衡群集:原理、分类与NAT模式实战部署
分类·负载均衡·lvs
Hcoco_me21 小时前
深度学习和神经网络之间有什么区别?
人工智能·深度学习·神经网络
astragin21 小时前
神经网络常见层速查表
人工智能·深度学习·神经网络
蒋星熠1 天前
深度学习实战指南:从神经网络基础到模型优化的完整攻略
人工智能·python·深度学习·神经网络·机器学习·卷积神经网络·transformer
victory04311 天前
wav2vec微调进行疾病语音分类任务
人工智能·分类·数据挖掘
大千AI助手1 天前
Dropout:深度学习中的随机丢弃正则化技术
人工智能·深度学习·神经网络·模型训练·dropout·正则化·过拟合
大霸王龙1 天前
基于vLLM与YOLO的智能图像分类系统
yolo·分类·数据挖掘