Ver Fonte

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

zhouyang.xie há 3 meses atrás
pai
commit
b17ab0381a
2 ficheiros alterados com 2 adições e 1 exclusões
  1. 0 0
      data/processed/train.jsonl
  2. 2 1
      src/generate_data.py

Diff do ficheiro suprimidas por serem muito extensas
+ 0 - 0
data/processed/train.jsonl


+ 2 - 1
src/generate_data.py

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

Alguns ficheiros não foram mostrados porque muitos ficheiros mudaram neste diff