自定义决策树保存——python pickle模块实现
定义两个函数,storeTree用于把决策树以二进制形式保存到文件中,grabTree从文件中读出决策树到内存文件后缀名为.pkl#!/usr/bin/env python# -*- coding: utf-8 -*-import pickledef storeTree(inputTree, filename):"""Serialize the decision t.
·
定义两个函数,storeTree用于把决策树以二进制形式保存到文件中,grabTree从文件中读出决策树到内存
文件后缀名为.pkl
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import pickle
def storeTree(inputTree, filename):
"""Serialize the decision tree and save it to a pickle file."""
fw = open(filename,'wb')
pickle.dump(inputTree,fw)
fw.close()
def grabTree(filename):
"""Convert the decision tree file into memory."""
fr = open(filename,'rb')
return pickle.load(fr)
注:这里打开方式得写成'wb'和'rb',否则会出现异常。
更多推荐
已为社区贡献4条内容
所有评论(0)