python 模型的保存

发布时间:2019-07-30 10:52:59编辑:auto阅读(1702)

    这里写图片描述

    来源于知乎

    模型上线一般通过java处理
    此时最好用pmml,github上有sklearntopmml的模块可以免费使用,强烈推荐。
    这和R语言有点类似

    完整的一个例子

    # conding = utf-8
    from sklearn import svm
    from sklearn.externals import joblib
    import os
    X = [[0,1],[0,2],[0,3],[1,0],[2,0],[3,0]]
    y= [0,0,0,1,1,1]
    clf = svm.SVC(probability=True)
    clf.fit(X,y)
    # 测试样本
    test = [[2.5,2.5],[2.5,0]]
    result = clf.predict_proba(test)
    print(result)
    
    joblib.dump(clf,"train_model.m")
    clf2 =joblib.load("train_model.m")
    test2 =[[2.,2.],[1.,2.]]
    result2 = clf2.predict_proba(test2)
    print(result2)
    

关键字