决策树算法优化(三)

来源:互联网 发布:淘宝模板代码怎么使用 编辑:程序博客网 时间:2024/05/17 07:42

在数据量很小的情况下,使用决策树能够很快的对数据进行分类,如果数据量大的话,构造决策树将耗费大量的时间,如果用创建好的决策树来进行分类能很好的解决这个问题。
使用pickle存储决策树,使用的时候直接读取。

def storeTree(inputTree,filename):    fw=open(filename,'wb')    pickle.dump(inputTree,fw)    fw.close()def grabTree(filename):    fr=open(filename,'rb')    data=pickle.load(fr)    fr.close()    return datastoreTree(mytree,'mytree.pkl')grabTree('mytree.pkl')
0 0
原创粉丝点击