我刚开始使用 keras,想在编译模型时使用未加权的 kappa 作为指标。按照此处的说明,我尝试按如下方式定义我的自定义指标:
library(DescTools) # includes function to calculate kappa
library(keras)
metric_kappa <- function(y_true, y_pred) {
CohenKappa(y_true, y_pred)
}
model %>% compile(
loss = 'categorical_crossentropy',
optimizer = optimizer_rmsprop(),
metrics = metric_kappa
)
但是,我的定义一定有问题,因为我遇到了分段错误。任何关于如何解决这个问题以及如何正确设置自定义指标的想法都将受到高度赞赏。