TypeError: write() argument must be str, not bytes

数据架构

于 2022-04-15 14:44:38 发布

阅读量472

点赞数

分类专栏： Maxwell‘s Bug

本文链接：https://blog.youkuaiyun.com/u011868279/article/details/124194329

版权

决策树 pickle 文件持久化优化二进制模式

关键词由优快云通过智能技术生成

Maxwell‘s Bug 专栏收录该内容

39 篇文章

订阅专栏

Original code:

# Methods for persisting the decision tree with pickle

def storeTree(inputTree, filename):
    import pickle
    fw = open(filename,'w')
    pickle.dump(inputTree,fw)
    fw.close()

def grabTree(filename):
    import pickle
    fr = open(filename)
    return pickle.load(fr)

Modified:

# Methods for persisting the decision tree with pickle

def storeTree(inputTree, filename):
    import pickle
    fw = open(filename,'wb+')
    pickle.dump(inputTree,fw)
    fw.close()

def grabTree(filename):
    import pickle
    fr = open(filename,'rb')
    return pickle.load(fr)