8 多输出预测与多标签分类pytorch网络搭建


前言

前面我们搭建的无论是分类还是回归都只能预测一个标签,这显然效果很局限。下面我们想做到下面这两种效果:

  • 多输出预测(回归):例如训练网络拟合北东天坐标转机体坐标的关系,输入是三坐标,输出也是三坐标
  • 多标签分类:例如,输入图像数据,训练网络判断图片里面有猫,有狗,还是只有其中一种这样

【注】:在介绍pytorch的内置损失函数博客中已经介绍了pytorch的损失函数是支持这个功能的。

一、多输出预测(回归)

1 坐标数据生成

# 本示例演示如何使用 PyTorch 实现多标签回归模型。
import torch
import torch.nn as nn
import torch.optim as optim
import numpy as np

# 构建数据集
# 假设您有一些经纬高度和对应的地心地固坐标的数据
# 这里只是一个示例,您需要根据实际情况准备您自己的数据集
X = np.random.rand(100, 3)  # 100个样本,每个样本有3个特征(经度、纬度、高度)
y = np.random.rand(100, 3)  # 每个样本有3个目标值(地心地固坐标)
print('y:\n',y)

在这里插入图片描述

2 网络搭建训练预测

# 转换数据为 PyTorch 的 Tensor 类型
X_tensor = torch.tensor(X, dtype=torch.float32)
y_tensor = torch.tensor(y, dtype=torch.float32)

# 定义模型
class MultiLabelRegressionModel(nn.Module):
    def __init__(self, input_size, output_size):
        super(MultiLabelRegressionModel, self).__init__()
        self.fc = nn.Linear(input_size, output_size)
        
    def forward(self, x):
        out = self.fc(x)
        return out

# 初始化模型
input_size = 3   # 输入特征的数量
output_size = 

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-17 00:22:02       66 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-17 00:22:02       70 阅读
  3. 在Django里面运行非项目文件

    2024-07-17 00:22:02       57 阅读
  4. Python语言-面向对象

    2024-07-17 00:22:02       68 阅读

热门阅读

  1. 1.3Zygote

    2024-07-17 00:22:02       20 阅读
  2. 精准打击:Conda中conda remove命令的高效使用指南

    2024-07-17 00:22:02       22 阅读
  3. react项目使用EventBus实现登录拦截

    2024-07-17 00:22:02       20 阅读
  4. MySQL 关键字 IN 与 EXISTS 的使用与区别

    2024-07-17 00:22:02       22 阅读
  5. 关于ARP欺骗

    2024-07-17 00:22:02       20 阅读
  6. 20. Python读取.mat格式文件通用函数

    2024-07-17 00:22:02       15 阅读
  7. wxml与标准的html的异同

    2024-07-17 00:22:02       17 阅读
  8. 优雅退出Conda环境:conda deactivate命令的妙用

    2024-07-17 00:22:02       21 阅读
  9. 原码补码反码移码

    2024-07-17 00:22:02       18 阅读
  10. Linux 多进程编程详解

    2024-07-17 00:22:02       20 阅读
  11. [Css-in-js]wyw-in-js的使用

    2024-07-17 00:22:02       20 阅读
  12. 学习编程的第一天

    2024-07-17 00:22:02       19 阅读