代码之家  ›  专栏  ›  技术社区  ›  AIM

R:具有自定义损失函数的深度神经网络

  •  1
  • AIM  · 技术社区  · 6 年前

    假设我有一个 loss function 并将其用作 DNN regression ? 为了进行深度学习,我目前正在使用 tensorflow -> R 接口。

    1 回复  |  直到 6 年前
        1
  •  1
  •   OzanStats    6 年前

    R的实现允许您使用自定义的loss函数。但是,该函数需要使用非常特定的语法来实现,并且应该采用 y_true y_pred here . 以下代码将为您提供一些直觉:

    model %>% compile(
      optimizer = "your-choice-of-optimezer",
      loss = custom_loss_function,
      metrics = c("your-choice-of-metric")
    )
    

    哪里

    custom_loss_function <- function(y_true, y_pred) {
      K <- backend()
      ... # define your function using the backend K
    }