训练模型保存和加载(sklearn)

    科技2023-11-16  97

    很多模型训练完成之后,可以进行保存,下次使用时直接调用即可,不需要再次训练数据。接下来我将介绍sklearn中模型的保存和加载。

    sklearn保存模型的api

    from sklean.externals import joblib #保存训练模型 joblib.dump(lr,"./tmp/test.pkl") #导入模型数据 lr2=joblib.load("./tmp/test.pkl")

    保存数据和导入数据都是训练好的模型,里面包含了模型的参数。

    例如:

    #模型 lr = LinearRegression() lr.fit(x_train, y_train) #保存模型 joblib.dump(lr,"./tmp/test.pkl") #导入模型 lr2=joblib.load("./tmp/test.pkl") lr2.fit(x_train, y_train)
    Processed: 0.013, SQL: 8