kaggle课程（二）Intro to Machine Learning

一、Your First Machine Learning Model

# Code you have previously used to load data
import pandas as pd

# 加载数据
iowa_file_path = '../input/home-data-for-ml-course/train.csv'

home_data = pd.read_csv(iowa_file_path)

#删除空值的行
home_data = home_data.dropna(axis=0)

#预览
home_data.columns

#选取目标预测值Specify Prediction Target
y = home_data.SalePrice

# 创建特征值Create the list of features below
feature_names = ['LotArea','YearBuilt','1stFlrSF','2ndFlrSF','FullBath','BedroomAbvGr','TotRmsAbvGrd']
# Select data corresponding to features in feature_names
X = home_data[feature_names]

# 预览Review data
# print description or statistics from X
print(X.describe())
# print the top few lines
print(X.head())

from sklearn.tree import DecisionTreeRegressor
#创建模型specify the model. 
#For model reproducibility, set a numeric value for random_state when specifying the model
iowa_model = DecisionTreeRegressor(random_state = 1)

# Fit the model
iowa_model.fit(X,y)

#Make Predictions
predictions = iowa_model.predict(X)
print(predictions)
prin