如何在训练后将新数据扩展到神经网络中

时间:2017-11-09 17:25:56

标签: python scikit-learn keras

我有一个训练有素的神经网络,但是我预先缩放了数据[sklearn preprocessing.scale(X)]。这很有效,但是当我保存模型并想要输入新数据时,我该怎么办?如何确保缩放与训练时相同?

selectedLocation

1 个答案:

答案 0 :(得分:1)

通常做的是保存scale对象,因为您可能会阅读here

以下是保存scale对象的方法:

from sklearn.externals import joblib
joblib.dump(scale, 'filename.pkl') 

以下是您可以重新加载它的方法:

scale = joblib.load('filename.pkl') 

在预测新数据时 - 最好重新加载缩放器 - 相应地缩放数据然后使用您的模型。你甚至可以创建一个新的类,其中保存了scaler并保存为keras.model