GatherPredsCallback: GatherPredsCallback

Description Usage Arguments Value

View source: R/callbacks.R

Description

'Callback' that saves the predictions and targets, optionally 'with_loss'

Usage

1
2
3
4
5
6
7
GatherPredsCallback(
  with_input = FALSE,
  with_loss = FALSE,
  save_preds = NULL,
  save_targs = NULL,
  concat_dim = 0
)

Arguments

with_input

include inputs or not

with_loss

include loss or not

save_preds

save predictions

save_targs

save targets/actuals

concat_dim

concatenate dimensions

Value

None


fastai documentation built on July 28, 2021, 5:06 p.m.