File size: 510 Bytes
5fae609 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import json
from pathlib import Path
from copy import copy
config = json.loads(Path("config.json").read_text())
code_map = json.loads(Path("code_map.json").read_text()) # label : ID
code_map_inv = {v: k for k, v in code_map.items()} # id : label
code_map_inv_str = {str(k): str(v) for k, v in code_map_inv.items()}
code_map_str = {str(v): str(k) for k, v in code_map.items()}
config["id2label"] = code_map_inv_str
config["label2id"] = code_map
Path("config.json").write_text(json.dumps(config, indent=4))
|