AccumMetric

Stores predictions and targets on CPU in accumulate to perform final calculations with `func`.

AccumMetric(
  func,
  dim_argmax = NULL,
  activation = "no",
  thresh = NULL,
  to_np = FALSE,
  invert_arg = FALSE,
  flatten = TRUE,
  ...
)

Arguments

func

function

dim_argmax

dimension argmax

activation

activation

thresh

threshold point

to_np

to matrix or not

invert_arg

invert arguments

flatten

flatten

...

additional arguments to pass

Value

None