生成sft数据,设置OQMD的代理,测试mars-t1
This commit is contained in:
13
generate_data/read_data.py
Executable file
13
generate_data/read_data.py
Executable file
@@ -0,0 +1,13 @@
|
||||
import copy
|
||||
import jsonlines
|
||||
|
||||
data_path = '/home/ubuntu/sas0/lzy/mars-mcp/generate_data/agent_questions_solution_turn5_ans_no_none.jsonl'
|
||||
#output_path = './agent_questions_solutions_qwq1.jsonl'
|
||||
with jsonlines.open(data_path, mode='r') as reader:
|
||||
datas = [line for line in reader if len(line['messages']) == 4]
|
||||
|
||||
print(datas[0])
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user