绘制由sklearn制成的回归器的3d图

debugcn 发表于 Dev

阿比雅

我一直在使用本教程来学习决策树学习，现在正试图了解它如何与高维数据集一起工作。

目前，我的回归器预测传递给它的（x，y）对的Z值。

import numpy as np 
import matplotlib.pyplot as plt 
from sklearn.tree import DecisionTreeRegressor 
from mpl_toolkits import mplot3d
dataset = np.array( 
[['Asset Flip', 100,100, 1000], 
['Text Based', 500,300, 3000], 
['Visual Novel', 1500,500, 5000], 
['2D Pixel Art', 3500,300, 8000], 
['2D Vector Art', 5000,900, 6500], 
['Strategy', 6000,600, 7000], 
['First Person Shooter', 8000,500, 15000], 
['Simulator', 9500,400, 20000], 
['Racing', 12000,300, 21000], 
['RPG', 14000,150, 25000], 
['Sandbox', 15500,200, 27000], 
['Open-World', 16500,500, 30000], 
['MMOFPS', 25000,600, 52000], 
['MMORPG', 30000,700, 80000] 
]) 
X = dataset[:, 1:3].astype(int) 
y = dataset[:, 3].astype(int) 
regressor = DecisionTreeRegressor(random_state = 0) 
regressor.fit(X, y)

我想使用3D图形对其进行可视化，但是我一直难以满足regressor.predict（）期望其输入为vs.matplotlib线框等程序期望其输入为的方式。结果，我无法使它们一起工作。

赛

试试这个，我没有安装所有软件包，所以我在google colab上进行了测试。让我知道这是否是您的期望。

from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
# to just see the prediction results of your data
#ax.scatter(X[:, 0], X[:, 1], regressor.predict(regressor.predict(X)), c='g')

samples = 10
xx, yy = np.meshgrid(np.linspace(min(X[:,0]), max(X[:,0]), samples), np.linspace(min(X[:,1]), max(X[:,1]), samples))
# to see the decision boundaries(not the right word for a decision tree regressor, I think)
ax.plot_wireframe(xx, yy, regressor.predict(np.hstack((xx.reshape(-1,1), yy.reshape(-1,1)))).reshape(xx.shape))
ax.set_xlabel('x-axis')
ax.set_ylabel('y-axis')
ax.set_zlabel('z-axis(predictions)')