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

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)]