当前位置 博文首页 > 向日葵的专属太阳:机器学习——模型保存

    向日葵的专属太阳:机器学习——模型保存

    作者:[db:作者] 时间:2021-08-13 22:04

    对于已经调好参数的模型,我们需要将其保存下来,可以使用下面两种方法。


    模型训练

    from sklearn.tree import DecisionTreeClassifier
    from sklearn import datasets
    
    X, y = datasets.load_iris(return_X_y=True)
    model = DecisionTreeClassifier()
    model.fit(X, y)
    

    sklearn.externals.joblib

    import joblib
    
    # 模型保存
    joblib.dump(model, 'DecisionTree.pkl')
    # 模型加载
    model = joblib.load('DecisionTree.pkl')
    

    pickle

    import pickle
    
    # 模型保存
    f = open('DecisionTree.pickle', 'wb')
    pickle.dump(model, f)
    f.close()
    
    # 模型加载
    f = open('DecisionTree.pickle', 'rb')
    model = pickle.load(f)
    f.close()
    

    joblib与pickle对比

    对于大数据而言,joblibpickle更加高效,但是joblib只能将对象存储在磁盘文件中,不能保存为字符串。

    cs
    下一篇:没有了