如何在CNTK中使用scikit-learn指标?

时间:2018-04-02 14:12:59

标签: python neural-network cntk

我希望使用matthews_corrcoef之类的分类指标作为使用CNTK构建的神经网络的指标。我现在能找到的方法是通过传递预测和标签来评估价值

matthews_corrcoef(cntk.argmax(y_true, axis=-1).eval(), cntk.argmax(y_pred, axis=-1).eval())

理想情况下,我希望在构建网络时将指标传递给培训师对象。 其中一种方法是创建自己的自定义指标并将其传递给培训师对象。尽管可能,但能够重用其他库中现有的度量标准会更好。

1 个答案:

答案 0 :(得分:0)

除非此指标已在CNTK中实施,否则无论CNTK需要什么格式,都要实现您自己的自定义“指标”功能,并将输入传递给scikit-learn的指标函数。