我使用TensorFlow Keras创建了一个模型,运行效果似乎还不错。然而,我的导师说除了已经计算的准确率和损失之外,计算马修斯相关系数会很有帮助。
我的模型与此教程中的代码非常相似(https://www.tensorflow.org/tutorials/keras/basic_classification),只是使用了一个更小的数据集。
是否有预构建的函数可用,还是我需要为每个测试获取预测并手动计算?
回答:
用于计算马修斯相关系数的预构建函数
sklearn.metrics.matthews_corrcoef(y_true, y_pred, sample_weight=None )
示例:
> from sklearn.metrics import matthews_corrcoef> y_true = [+1, +1, +1, -1]> y_pred = [+1, -1, +1, +1]> matthews_corrcoef(y_true, y_pred)
请参阅文档。