【发布时间】:2020-03-07 06:58:08
【问题描述】:
我正在使用验证拆分训练和拟合 keras 模型:
self.model = Sequential()
self.model.add(LSTM(hidden_units, input_shape=(1, n_features), dropout=drp))
self.model.add(Dense(n_classes, activation='sigmoid'))
self.model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
self.model.build(X_train.shape)
self.history = self.model.fit(X, y.values, epochs=epochs, batch_size=32, validation_split=0.33, shuffle=True, callbacks=cb_list)
拟合后,我想访问使用的测试集。 我该怎么做?
【问题讨论】:
-
您应该提供更多信息。你用的是什么框架?标签上写着 scikit-learn,但你显示的方法签名让我觉得你正在使用 keras 或其他一些深度学习框架。
-
@xibalba1 请看我的编辑
-
您没有在上面的模型中使用测试。另外我认为你的意思是写
self.model.fit(X, y.values, epochs=epochs, batch_size=32, validation_split=0.33, shuffle=True, callbacks=cb_list)X = X_train和y.values = y_train。 -
@Philip 是的,这将 X,y 拆分为 0.66 / 0.33 对吗?那么如何访问验证集呢?
-
据我所知documentation,您可以访问history.history['acc'],history.history['val_acc'],history.history['loss'], history.history['val_loss']。但是,我在测试集上没有看到任何提及。但是,您可以在此操作之前拆分为测试和训练
标签: python machine-learning keras scikit-learn data-science