HF_SummarizationModelCallback: HF_SummarizationModelCallback

View source: R/blurr_hugging_face.R

HF_SummarizationModelCallbackR 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


fastai documentation built on March 21, 2022, 9:07 a.m.