我們再建模完畢以後,並不需要每次都把資料重新執行,而是將資料模型給保留下來
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...