codemafia0000 commited on
Commit
21e55ce
·
verified ·
1 Parent(s): 8f3737b

update readme

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -49,6 +49,9 @@ import json
49
 
50
  # モデル名の指定
51
  model_name = "codemafia0000/llm-jp-3-13b-finetune-LoRA-241124"
 
 
 
52
 
53
  # Hugging Faceのアクセストークンを設定します。自身のトークンに置き換えてください。
54
  HF_TOKEN = "YOUR_TOKEN"
@@ -104,7 +107,7 @@ for dt in tqdm(datasets):
104
  results.append({"task_id": dt["task_id"], "input": input, "output": prediction})
105
 
106
  # 結果をJSONL形式で保存
107
- with open(f"{model_name}_output.jsonl", 'w', encoding='utf-8') as f:
108
  for result in results:
109
  json.dump(result, f, ensure_ascii=False)
110
  f.write('\n')
 
49
 
50
  # モデル名の指定
51
  model_name = "codemafia0000/llm-jp-3-13b-finetune-LoRA-241124"
52
+ # 提出用ファイル名
53
+ output_file_name = model_name.split('/')[-1] # モデル名のみ取得
54
+ output_file_name = f"{output_file_name}_output.jsonl"
55
 
56
  # Hugging Faceのアクセストークンを設定します。自身のトークンに置き換えてください。
57
  HF_TOKEN = "YOUR_TOKEN"
 
107
  results.append({"task_id": dt["task_id"], "input": input, "output": prediction})
108
 
109
  # 結果をJSONL形式で保存
110
+ with open(output_file_name, 'w', encoding='utf-8') as f:
111
  for result in results:
112
  json.dump(result, f, ensure_ascii=False)
113
  f.write('\n')