小编Ren*_*huk的帖子

我很好奇valueGraphviz用于回归时决策树的节点中的字段是什么.我知道这是使用决策树分类时每个类中由分割分隔的样本数,但我不确定它对回归意味着什么.

我的数据有2维输入和10维输出.以下是我的回归问题树的示例:

使用此代码生成并使用webgraphviz进行可视化

# X = (n x 2)  Y = (n x 10)  X_test = (m x 2)

input_scaler = pickle.load(open("../input_scaler.sav","rb"))
reg = DecisionTreeRegressor(criterion = 'mse', max_depth = 2)
reg.fit(X,Y)
pred = reg.predict(X_test)
with open("classifier.txt", "w") as f:
    f = tree.export_graphviz(reg, out_file=f)

谢谢!