基于pytorch写一个多对多预测代码,数据集名称为data.csv,前四列为特征,后面50列为标签,前800行为训练集其余为测试集,将其中一组预测值与真实值输出到excel工作表中...
下面是基于 PyTorch 实现的一个多对多分类预测代码示例:
import torch
import pandas as pd
# 读入数据
data = pd.read_csv("data.csv")
X_train, y_train = data.iloc[:800, :4], data.iloc[:800, 4:]
X_test, y_test = data.iloc[800:, :4], data.iloc[800:, 4:]
# 转换为 PyTorch 张量
X_train, y_train = torch.tensor(X_train.values),