# 使用Pickle包,读取文件数据,使其转换成Python数据
def saveDecisionTree(decisionTree,filename):
'''
:param decisionTree: 已生成的决策树
:param filename: 文件路径
:return:
:function: 保存决策树
'''
import pickle
fw = open(filename,'wb')
pickle.dump(decisionTree,fw)
fw.close()
def readDecisionTree(filename):
'''
:param filename: 文件路径
:return: 决策树
:function: 读取决策树
'''
import pickle
fr = open(filename,'rb')
return pickle.load(fr)
01-20