【PyTorch】MLP回归模型预测波士顿房价

最新推荐文章于 2024-07-28 22:59:25 发布

12smile25

最新推荐文章于 2024-07-28 22:59:25 发布

阅读量7.1k

点赞数 7

分类专栏：神经网络文章标签：深度学习 python

本文链接：https://blog.csdn.net/weixin_39274067/article/details/111154355

版权

import numpy as np
import pandas as pd
from sklearn.preprocessing import StandardScaler
from sklearn.metrics import mean_squared_error,mean_absolute_error
from sklearn.model_selection import train_test_split
from sklearn.datasets import fetch_california_housing
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.optim import SGD
import torch.utils.data as Data
import matplotlib.pyplot as plt
import seaborn as sns

#导入数据
housedata=fetch_california_housing()
#划分测试集和训练集
X_train,X_test,y_train,y_test=train_test_split(housedata.data,housedata.target,test_size=0.3,random_state=42)
#标准化处理
scale=StandardScaler()
X_train_s=scale.fit_transform(X_train)
X_test_s=scale.fit_transform(X_test)
housedatadf=pd.DataFrame(data=X_train_s,columns=housedata.feature_names)
housedatadf["target"]=y_train

#可视化训练数据的相关系数热力图 8个自变量以及目标之间的相关系数
datacor=np.corrcoef(housedatadf.values,rowvar=0)
datacor=pd.DataFrame(data=datacor,columns