GrammarErrorCorrectionTemplate

class torchtune.data.GrammarErrorCorrectionTemplate

Prompt template for grammar correction datasets.

classmethod format(sample: Mapping[str, Any], column_map: Optional[Dict[str, str]] = None) → str

Generate prompt from sentence.

Parameters:

• sample (Mapping[str, Any]) – a single data sample with sentence

• column_map (Optional[Dict[str, str]]) – a mapping from the expected placeholder names in the template to the column names in the sample. If None, assume these are identical.

Returns:

The formatted prompt