【发布时间】:2017-11-09 17:25:55
【问题描述】:
我有一个训练有素的神经网络,效果非常好,但我事先对数据进行了缩放 [sklearn preprocessing.scale(X)]。这很好用,但是当我保存模型并想向其中输入新数据时该怎么办?如何确保缩放与训练时相同?
df = pd.read_csv("Trimmed Training Data.csv", delimiter=",")
X = np.array(df.drop(['PredictThis'],1))
y = np.array(df['PredictThis'])
X = preprocessing.scale(X)
# create NN model
model = Sequential()
# 2 inputs, 10 neurons in 1 hidden layer, with tanh activation and dropout
model.add(Dropout(0.2, input_shape=(15,)))
model.add(Dense(100, init='uniform', input_shape=(15,), activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(100, init='uniform', input_shape=(100,), activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(5, init='uniform', input_shape=(100,), activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(1, init='uniform', activation='sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='rmsprop')
# Fit the model
model.fit(X, y, nb_epoch=25, batch_size=15)
【问题讨论】:
-
您可能应该找出预处理使用的缩放比例,并使用相同的参数执行您自己的手动缩放。
-
而且 sklearn 提供了更通用的基于类的缩放器,您可以保存并重复使用以供以后缩放。您使用的便捷功能不包含所需的信息。
标签: python scikit-learn keras