上一篇讲述了决策树分类模型
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")