Kaggle 学习之旅

决策树

https://www.kaggle.com/dansbecker/your-first-machine-learning-model

import pandas as pd  
melb_data_path = 'melb_data.csv'
data1 = pd.read_csv(melb_data_path)
data1.describe()
data1.columns
data2 = data1.dropna(axis=0)
data2.describe()
y = data2.Price #定义target
y.describe()
features = ['Rooms', 'Bathroom', 'Landsize', 'Lattitude', 'Longtitude']
X = data2[features] #定义feature
X.describe()
X.head()

from sklearn.tree import DecisionTreeRegressor
model1 = DecisionTreeRegressor(random_state=1) #选择决策树模型
model1.fit(X,y)  #训练模型
X.head()
model1.predict(X.head()) #使用模型对X样本前5行进行价格预测

output：

>>> X.head()

   Rooms  Bathroom  Landsize  Lattitude  Longtitude

1      2       1.0     156.0   -37.8079    144.9934

2      3       2.0     134.0   -37.8093    144.9944

4      4       1.0     120.0   -37.8072    144.9941

6      3       2.0     245.0   -37.8024    144.9993

7      2       1.0     256.0   -37.8060    144.9954

>>> model1.predict(X.head())

array([1035000., 1465000., 1600000., 1876000., 1636000.])

 

 

其他：

查看python 历史命令：    import readline; print '\n'.join([str(readline.get_history_item(i + 1)) for i in range(readline.get_current_history_length())])