Jelajahi Sumber

修改 风电机组数据集格式

zhouyang.xie 2 bulan lalu
induk
melakukan
948ec7eb5d
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  1. 2 2
      src/generate_data.py

+ 2 - 2
src/generate_data.py

@@ -35,8 +35,8 @@ class DataGenerator:
                 "answer":   "".join([case_data[2],"\n",case_data[3],"\n",case_data[4]]),
                 "prompt": [
                     {
-                        "content": f"\nRespond in the following format:\n<reasoning>\n {case_data[2]} \n</reasoning>\n<answer>\n {case_data[3]}  {case_data[4]} \n</answer>\n",
-                        # "content": f"\nRespond in the following format:\n<reasoning>\n ... \n</reasoning>\n<answer>\n ... \n</answer>\n",
+                        # "content": f"\nRespond in the following format:\n<reasoning>\n {case_data[2]} \n</reasoning>\n<answer>\n {case_data[3]}  {case_data[4]} \n</answer>\n",
+                        "content": "\nRespond in the following format:\n<reasoning>\n...\n</reasoning>\n<answer>\n...\n</answer>\n",
                         "role": "system"
                     },
                     {