We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b961717 commit 0451b92Copy full SHA for 0451b92
src/setfit/trainer_distillation.py
@@ -45,7 +45,7 @@ class DistillationTrainer(Trainer):
45
column_mapping (`Dict[str, str]`, *optional*):
46
A mapping from the column names in the dataset to the column names expected by the model.
47
The expected format is a dictionary with the following format:
48
- `{"text_column_name": "text", "label_column_name: "label"}`.
+ `{"text_column_name": "text", "label_column_name": "label"}`.
49
"""
50
51
_REQUIRED_COLUMNS = {"text"}
0 commit comments