我正在尝试使用现有代码https://github.com/mgckind/MLZ/blob/master/mlz/ml_codes/TPZ.py在Python中构建回归树。特别是代码中的Rtree()。
This is the code我正在运行构建一个调用TPZ程序的树。 当我调用T.plot_tree()时,我遇到问题的是绘制树。 This is the error我在尝试运行程序时得到了。该代码的作者说它是为Linux环境构建的。所以我在windows中安装了一个linux环境(使用/ lxrun方法)但是在编码方面是一个门外汉,我真的无法解决这个问题。谁能帮我?
如果信息不充分,请道歉,如果您还有其他需要,请告诉我。
谢谢!
编辑:这是我使用的代码。我之前遇到的最初问题是没有找到正确的道路。然后我在windows中安装了linux环境,并且能够解决这个问题。但现在问题似乎在于阅读PNG文件。我是如何为此安装python包Matplotlib的?我希望这有帮助!
from numpy import *
import time
import os, sys
path_src = os.path.abspath(os.path.join(os.getcwd(), '../../'))
if not path_src in sys.path: sys.path.insert(1, path_src)
from ml_codes import *
start = time.time()
X = loadtxt('Data/Query3.txt', usecols=(1,2,3,4,5,6,7), unpack=True).T
Y = loadtxt('Data/Query3.txt', unpack=True, usecols=(8))
d = {'u': {'ind': 0}, 'g': {'ind': 1}, 'r': {'ind': 2}, 'i': {'ind': 3},
'z': {'ind': 4}, 'u-g': {'ind': 5},
'g-r': {'ind': 6}}
T = TPZ.Rtree(X, Y, minleaf=1,forest = 'yes', mstar=4, dict_dim = d)
T.plot_tree()
branches = T.leaves()
att = T.leaves_dim()
print 'tree structure = ', branches
print 'best split at each node = ', att