【机器学习】决策树回归模型+tips 数据集实战

上一篇讲述了决策树分类模型

1、决策树回归模型

from sklearn.tree import DecisionTreeRegressor
dr=DecisionTreeRegressor(max_depth=3)深度为3

2、

import numpy as np
import pandas as pd
from pandas import Series,DataFrame
import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline
tip=sns.load_dataset('tips')
tip
# tip.shape=(244, 7)
# type(tip)=pandas.core.frame.DataFrame

train=tip[["total_bill"]]
target=tip.tip
from sklearn.tree import DecisionTreeRegressor
dr=DecisionTreeRegressor(max_depth=3)
dr.fit(train,target)
# test.shape=(50, 1)
test=np.linspace(train.min(),train.max(),100).reshape(-1,1)
y_test_p=dr.predict(test)
plt.scatter(tip.total_bill,tip.tip)
plt.plot(test,y_test_p,color="green")

 

完整代码：

import numpy as np
import pandas as pd
from pandas import Series,DataFrame
import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline
tip=sns.load_dataset('tips')
train=tip[["total_bill"]]
target=tip.tip
from sklearn.tree import DecisionTreeRegressor
dr=DecisionTreeRegressor(max_depth=3)
dr.fit(train,target)
test=np.linspace(train.min(),train.max(),100).reshape(-1,1)
y_test_p=dr.predict(test)
plt.scatter(tip.total_bill,tip.tip)
plt.plot(test,y_test_p,color="green")