HF_SummarizationModelCallback {fastai}R Documentation

HF_SummarizationModelCallback

Description

Basic class handling tweaks of the training loop by changing a 'Learner' in various events

Usage

HF_SummarizationModelCallback(
  rouge_metrics = c("rouge1", "rouge2", "rougeL"),
  ignore_token_id = -100,
  ...
)

Arguments

rouge_metrics

rouge metrics

ignore_token_id

integer, ignore token id

...

additional arguments

Value

None


[Package fastai version 2.2.2 Index]