使用后获取列车/测试数据型号.fit()带验证\u spli

2024-05-16 13:21:10 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用验证拆分对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)

安装之后,我想访问所使用的测试集。 我该怎么做?你知道吗


Tags: 模型selfaddinputmodelhiddendropoutkeras
1条回答
网友
1楼 · 发布于 2024-05-16 13:21:10

根据documentation中的FAQ,validation_split属性将最后%的样本作为验证数据。你知道吗

在本例中,您将validation_split的值设置为0.33。这意味着X中最后33%的样本被用作验证数据。你知道吗

因此,您可以直接切掉X的最后33%并使用它。你知道吗

相关问题 更多 >