InputOutputToMessages

class torchtune.data.InputOutputToMessages(train_on_input: bool = False, column_map: Optional[Dict[str, str]] = None, new_system_prompt: Optional[str] = None)

消息转换类，将具有“input”和“output”字段（或在 column_map 中指定的等效字段）的单个样本分别转换为用户和助手消息。这对于具有两列的数据集很有用，一列包含用户提示字符串，另一列包含模型响应字符串

|  input          |  output          |
|-----------------|------------------|
| "user prompt"   | "model response" |

参数：

• train_on_input (bool) – 模型是否在用户提示上进行训练。默认为 False。

• column_map (Optional[Dict[str, str]]) – 用于将预期的“input”和“output”列名称更改为数据集中实际列名称的映射。键应为“input”和“output”，值应为实际列名称。默认为 None，保留默认的“input”和“output”列名称。

• new_system_prompt (Optional[str]) – 如果指定，则在前面添加系统消息。这可以作为指导模型响应的说明。默认为 None。

引发：

ValueError – 如果提供 column_map 且 input 不在 column_map 中，或 output 不在 column_map 中。