Keras-3-实例2-多分类问题

简介: Keras-3-实例2-多分类问题

在这个例子中,我们将使用Keras处理多分类问题。我们将使用Iris数据集,该数据集包含三个不同种类的鸢尾花,每个种类包含50个样本。每个数据点都有四个特征:萼片长度、萼片宽度、花瓣长度和花瓣宽度。

我们将使用一个简单的神经网络来训练模型,并使用softmax激活函数输出一个概率分布,即每个样本属于三个类别中的哪一个。我们还将在训练过程中使用交叉熵损失函数和随机梯度下降优化器。

代码示例:

载入数据集

from sklearn.datasets import load_iris
iris = load_iris()

将数据集拆分为训练集和测试集

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.2, random_state=42)

将标签转换为one-hot编码

from keras.utils import to_categorical
y_train = to_categorical(y_train)
y_test = to_categorical(y_test)

创建模型

from keras.models import Sequential
from keras.layers import Dense

model = Sequential()
model.add(Dense(units=10, input_dim=4, activation='relu'))
model.add(Dense(units=3, activation='softmax'))

编译模型

model.compile(loss='categorical_crossentropy', optimizer='sgd', metrics=['accuracy'])

训练模型

model.fit(X_train, y_train, epochs=50, batch_size=10)

评估模型

loss_and_metrics = model.evaluate(X_test, y_test, batch_size=10)
print(loss_and_metrics)
python

我们首先将数据集拆分为训练集和测试集,然后使用to_categorical函数将标签转换为one-hot编码。

接下来,我们创建一个Sequential模型,并为其添加两个Dense层。第一层包含10个神经元和relu激活函数,第二层包含3个神经元和softmax激活函数,以输出每个样本属于三个类别中的哪一个的概率分布。我们使用交叉熵损失函数和随机梯度下降优化器来编译模型,并指定metrics为accuracy。

我们使用fit函数训练模型,并在测试集上使用evaluate函数评估模型的性能。

相关文章
|
12月前
|
费德勒权变模型(Fiedler Contingency Model)详解与Python代码示例
费德勒权变模型(Fiedler Contingency Model)详解与Python代码示例
DNS解析问题之授权RAM子账号管理指定域名如何解决
DNS解析是指将人类可读的域名转换成机器可读的IP地址的过程,它是互联网访问中不可或缺的一环;本合集将介绍DNS解析的机制、类型和相关问题的解决策略,以确保域名解析的准确性和高效性。
649 1
对象存储oss使用问题之C++使用OSS SDK时遍历OSS上的文件时崩溃如何解决
《对象存储OSS操作报错合集》精选了用户在使用阿里云对象存储服务(OSS)过程中出现的各种常见及疑难报错情况,包括但不限于权限问题、上传下载异常、Bucket配置错误、网络连接问题、跨域资源共享(CORS)设定错误、数据一致性问题以及API调用失败等场景。为用户降低故障排查时间,确保OSS服务的稳定运行与高效利用。
513 0
找不到中文语音预训练模型?中文版 Wav2vec 2.0和HuBERT来了,腾讯游戏知几AI团队和西工大ASLP组联合出品
找不到中文语音预训练模型?中文版 Wav2vec 2.0和HuBERT来了,腾讯游戏知几AI团队和西工大ASLP组联合出品
718 0
【Agones系列】Game Server的扩缩容
本文介绍了Agones中GameServer是如何进行扩缩容的
kde
|
3天前
|
Docker镜像加速指南:手把手教你配置国内镜像源
配置国内镜像源可大幅提升 Docker 拉取速度,解决访问 Docker Hub 缓慢问题。本文详解 Linux、Docker Desktop 配置方法,并提供测速对比与常见问题解答,附最新可用镜像源列表,助力高效开发部署。
kde
2073 6
2025年最新版最细致Maven安装与配置指南(任何版本都可以依据本文章配置)
本文详细介绍了Maven的项目管理工具特性、安装步骤和配置方法。主要内容包括: Maven概述:解释Maven作为基于POM的构建工具,具备依赖管理、构建生命周期和仓库管理等功能。 安装步骤: 从官网下载最新版本 解压到指定目录 创建本地仓库文件夹 关键配置: 修改settings.xml文件 配置阿里云和清华大学镜像仓库以加速依赖下载 设置本地仓库路径 附加说明:包含详细的配置示例和截图指导,适用于各种操作系统环境。 本文提供了完整的Maven安装和配置
2025年最新版最细致Maven安装与配置指南(任何版本都可以依据本文章配置)
Dify MCP 保姆级教程来了!
大语言模型,例如 DeepSeek,如果不能联网、不能操作外部工具,只能是聊天机器人。除了聊天没什么可做的。
575 5
国内如何安装和使用 Claude Code镜像教程 - Windows 用户篇
国内如何安装和使用 Claude Code镜像教程 - Windows 用户篇
376 0
AI助理

你好,我是AI助理

可以解答问题、推荐解决方案等

登录插画

登录以查看您的控制台资源

管理云资源
状态一览
快捷访问