HF_SummarizationModelCallback: HF_SummarizationModelCallback

Description Usage Arguments Value

View source: R/blurr_hugging_face.R

Description

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

Usage

1
2
3
4
5
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


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