with open("corpus.jsonl","r") as rd:
for line in tqdm(rd.readlines()):
line=line.strip()
obj=json.loads(line)
_id=obj["id"]
id2ent[_id]=obj
这个代码怎么写能让他跑得更快
with open("corpus.jsonl","r") as rd:
for line in tqdm(rd.readlines()):
line=line.strip()
obj=json.loads(line)
_id=obj["id"]
id2ent[_id]=obj
这个代码怎么写能让他跑得更快
|
1
|
|
2
for line in rd:
line = line.strip('\n') |
|
3
readlines 会一次性把所有行读进去,换逐行读取。
|
|
4
@killva4624 全读内存里不是会更快吗
|
|
LindsayZhouSep 19, 2022
6
如果性能瓶颈在 json 解析的话,用 ujson: https://github.com/ultrajson/ultrajson
看 fastapi 的时候找到的库。 |
|
renmu123Sep 20, 2022
7
起多进程,然后内容平分给每个进程来处理
|