前文我们使用机器学习法实现了鸢尾花的分类,
?
本文介绍使用Pytorch中的交叉熵算法来实现分类问题。
构建出的模型类型大致为
?
上图构建的输入层+2个隐藏层+输出层,共计4层结构的神经网络。...因此是4->layer1->layer2->3的三分类问题。考虑可以使用多种算法进行分析,本文先介绍使用BP神经网络进行分析。..., 2, 2, 2, 2, 2, 2, 2, 2, 2, 2])
引入pytorch工具包 构建BP网络
import torch.nn.functional as Fun
# 定义BP神经网络...prediction.numpy()
# 预测y输出数列
target_y = label.data.numpy()
# 实际y输出数据
输出正确率
正确率为: 0.98
该正确率明显高于前面的机器学习预测的正确率...后续还将介绍使用其他方法进行预测。