【发布时间】:2018-11-02 19:10:27
【问题描述】:
我有一个模型并使用以下方法对其进行拟合:
model.fit(Xtrain, ytrain, epochs = epochs, verbose = 2, batch_size = 1000, shuffle = True, validation_data = (Xtest, ytest))
但是,当使用内置的 validation_data 选项时,我无法找到一种方法来提取验证集的概率值和类别分配。
我正在考虑通过回调来实现这一点,类似于https://github.com/keras-team/keras/issues/2548 的情况,但我不确定该怎么做。
【问题讨论】:
标签: python python-3.x keras