機器學習與深度學習 模型的儲存和載入

2021-10-03 04:28:56 字數 1151 閱讀 2877

我們再建模完畢以後,並不需要每次都把資料重新執行,而是將資料模型給保留下來

from sklearn.externals import joblib

#儲存:

joblib.dump(estimator,

'test.m'

)#載入:

estimator = joblib.load(

'test.m'

)

詳細案例:

import numpy as np

import pandas as pd

import pickle as pk

from sklearn.externals import joblib

cancer=pd.read_csv(

'cancer.csv'

,sep=

'\t'

)

fea = cancer.select_dtypes(include=

['float64'])

lab = cancer.select_dtypes(include=

['object'

])

#載入模型

with

open

('best_knn.m'

,'rb'

)as fp:

best_knn = pk.load(fp)

with

open

('std_model.m'

,'rb'

)as fp:

std = pk.load(fp)

std=joblib.load(filename=

'std_model.m'

)

best_knn=joblib.load(filename=

'best_knn.m'

)

fea_std = std.transform(fea)
best_knn.score(fea_std,lab)

機器學習模型儲存

在訓練完成機器學習模型後,經常將滿足需要的機器學習模型進行儲存,本文以svm演算法為例,講解模型儲存和呼叫的方法。joblib.dump 模型,模型命名 將訓練模型儲存起來如 from sklearn.svm import svc from sklearn.metrics import classi...

機器學習 模型儲存

目錄 pickle模組 sklearn joblib模組 hive分割槽表 在訓練模型後將模型儲存的方法,以免下次重複訓練。以及大資料裡直接將bi模型結果儲存在hive表裡。from sklearn import svm from sklearn import datasets import pic...

機器學習模型的儲存與呼叫

在機器學習中常常會用到將訓練好的模型儲存,以便於在使用時不用再次訓練資料,就可以直接使用訓練好的模型。以sklearn中隨機森林分類為例 載入資料集 from sklearn.datasets import load breast cancer import pandas as pd from sk...