Notes
0%

XGBoost-Save

发表于 更新于 分类于 Valine：
XGBoost模型保存的两种方法

XGBoost保存模型的两个方法

pickle

1
2
3
4
5
6
7
8
9
10
11
12
from sklearn import svm
from sklearn import datasets
clf = svm.SVC(gamma='scale')
iris = datasets.load_iris()
X, y = iris.data, iris.target
clf.fit(X, y)  


import pickle
s = pickle.dumps(clf)
clf2 = pickle.loads(s)
clf2.predict(X[0:1])

joblib

1
2
3
4
from joblib import dump, load
dump(clf, 'filename.joblib') 

clf = load('filename.joblib')