#1 机器学习房价预测kaggle

首先,机器学习建模过程可以分为一下几个步骤

1.读取数据

2.数据的预处理

3.特征处理

4.划分数据

5.训练模型

6.评估模型

import matplotlib.pyplot as plt
import matplotlib
import seaborn as sns
import numpy as np
import pandas as pd
import warnings
warnings.filterwarnings("ignore")
#读入数据
test = pd.read_csv('test.csv')
train = pd.read_csv('train.csv')
sample_submission = pd.read_csv("sample_submission1.csv")

#展示数据
train.head() 
test.head()
#提取目标房价以及测试集id
target = train.SalePrice.values
test_ids = test.Id.values
#整理数据,去除要算的房价
train.drop(['SalePrice','Id'], axis=1 , inplace=True)
test.drop(["Id"],axis = 1,inplace = True)
#训练集长度

num_train = len(train)

#按照列拼接
data = pd.concat([train,test],axis=0).reset_index(drop=True) 
data.dtypes.value_counts()#看是不是数值
data.drop(data.isnull().mean()[data.isnull().mean() > 0.8].index.tolist(), axis=1, inplace=True) #处理数据
data.loc[:, data.dtypes[data.dtypes != "object"].index] = \
data.loc[:, data.dtypes[data.dtypes != "object"].index].fillna(
data.loc[:, data.dtypes[data.dtypes != "object"].index].median()) #数值填充
data.loc[:, data.dtypes[data.dtypes == "object"].index] = \
data.loc[:, data.dtypes[data.dtypes == "object"].index].fillna("NULL")#非数值填充
data.shape
data.head()


#sns.displot(target)
#sns.displot(np.log1p(target))

from sklearn.preprocessing import OneHotEncoder
from sklearn.preprocessing import StandardScaler
from sklearn.linear_model import LinearRegression, Ridge, Lasso
from sklearn.metrics import mean_squared_error as mse
cat_feature = OneHotEncoder().fit_transform(data[data.columns[data.dtypes == "object"]])  #特征数据选取
cat_feature.shape
num_feature = StandardScaler().fit_transform(data[data.columns[data.dtypes != "object"]]) #转化为数值数据 平均值=0 标准差=1
feature = np.concatenate([cat_feature.todense(), num_feature], axis=1)
#区分测试集以及训练集
test_feature = feature[num_train:] 
train_feature = feature[:num_train]

#归一化
target_log1p = np.log1p(target)
scaler = StandardScaler()
scaler.fit(target_log1p.reshape(-1, 1))

target_log1p_scale = scaler.transform(target_log1p.reshape(-1, 1))  #任意行 1列
target_log1p_scale.mean()


model = Ridge()#ridge回归

model.fit(train_feature, target_log1p_scale)

def predict (model,feature):
    pred = model.predict(feature)
    pred = scaler.inverse_transform(pred) #还原归一化
    pred = np.expm1(pred)#还原数据
    return pred

train_pred = predict(model, train_feature)
train_pred

np.sqrt(mse(target, train_pred)) #误差函数
test_pred = predict(model, test_feature) #训练模型

sample_submission["SalePrice"] = test_pred.reshape(-1)
sample_submission.to_csv("submission.csv1", index=False)

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值