使用 Qwen2.5B + QLoRA 微調情緒分類模型:訓練流程與評估結果實作
下面將每筆資料透過 _to_chat_text(...) 組成 ChatML 風格輸入
def _to_chat_text(tokenizer, instruction, user_input, output=None):
"""組成 ChatML 訓練 / 推理文本。output 為 None 時只到 assistant 起始。"""
messages = [{"role": "user", "content": f"{instruction}\n\n文本:{user_input}"}]
if output is not None:
messages.append({"role": "assistant", "content": output})
return tokenizer.apply_chat_template(messages, tokenize=False)
return tokenizer.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True
)
