在Scikit学习中重用定标器和估计器的持久性

时间:2019-04-05 02:38:45

标签: scikit-learn

由于我应该适合并转换x_train,但仅转换x_test,因此我试图保存缩放器和分类器,以备将来在不同的测试集上重用。 我不确定自己是否做对了。最好的方法是什么?

#Train and save
scaler = StandardScaler()
scaler.fit(x_train)
joblib.dump(scaler, 'scaler.joblib')
x_train = scaler.transform(x_train)
clf.fit(x_train, y_train)
joblib.dump(clf, 'clf.joblib')

#Load and test
clf = joblib.load('clf.joblib')
scaler = joblib.load('scaler.joblib')
x_test= scaler.transform(x_test)
preds = clf.predict(x_test)

0 个答案:

没有答案