代码之家  ›  专栏  ›  技术社区  ›  Jack Arnestad

使用验证数据时提取分类路缘石的概率值

  •  2
  • Jack Arnestad  · 技术社区  · 7 年前

    我有一个模型,并使用以下方法进行拟合:

    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 ,但我不确定怎么做。

    1 回复  |  直到 7 年前
        1
  •  0
  •   Akash Srivastava    7 年前

    试用使用 predict_prob

    你也可以找到同样的问题 stackexchange