瀏覽代碼

重新生成 风电机组故障诊断的数据集

zhouyang.xie 3 月之前
父節點
當前提交
2a0bb731cc
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      src/train_model_grpo_v1.1.py

+ 1 - 0
src/train_model_grpo_v1.1.py

@@ -364,6 +364,7 @@ class ModelTrainer:
         responses = [completion[0]['content'] for completion in completions]
         q = prompts[0][-1]['content']
         extracted_responses = [ModelTrainer.extract_xml_answer(r) for r in responses]
+        print(f"\n Response:\n {responses}",f"\n Extracted:\n {responses}")
         print('-' * 20, f"Question:\n{q}", f"\nAnswer:\n{answer[0]}", f"\nResponse:\n{responses[0]}", f"\nExtracted:\n{extracted_responses[0]}")
         return [2.0 if r == a else 0.0 for r, a in zip(extracted_responses, answer)]