深度学习BPNN落地实战代码

# baiyuntian
# @time: 2023/3/12 15:11
# @Author: Yang Bai
# @File: BPNN
# @Contact: 287503957@qq.com
# @Software: PyCharm
# baiyang
import numpy as np
import pandas as pd
import torch
import torch.nn as nn
from sklearn.preprocessing import MinMaxScaler
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score
import matplotlib.pyplot as plt

# 读取数据
data = pd.read_excel('data522.xlsx')

# 数据预处理
scaler = MinMaxScaler()
data_scaled = scaler.fit_transform(data)
data_normalized = pd.DataFrame(data_scaled, columns=data.columns)

# 提取"速度"列
speed = data_normalized['总推力']

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(data_normalized.drop('总推力', axis=1), speed, test_size=0.2,
                                                    random_state=42)

# 转换为PyTorch张量