AlpacaToMessages

class torchtune.data.AlpacaToMessages(train_on_input: 布尔值 = True, column_map: 可选值[字典[字符串, 字符串]] = None)

适用于具有“instruction”、“input”和“output”（或在 column_map 中指定的等效字段）列的 Alpaca 风格数据集的消息转换类。用户消息由 instruction + input 列构成，助手消息由 output 列构成。提示模板的生成取决于“input”列是否存在，因此由于此自定义逻辑，它在此转换类中直接处理，而不是在专用的 PromptTemplate 类中处理。

参数:

• train_on_input (布尔值) – 模型是否基于用户提示进行训练。默认为 True。

• column_map (可选值[字典[字符串, 字符串]]) – 用于将预期的“instruction”、“input”和“output”列名更改为数据集中实际列名的映射。默认为 None，保留默认列名。