xl2533 commited on
Commit
1f5a1b1
1 Parent(s): c282862
Files changed (1) hide show
  1. ape/instance.py +1 -1
ape/instance.py CHANGED
@@ -108,7 +108,7 @@ def load_entity(file='./ape/data/entity_train.json'):
108
  for j in i['labels']:
109
  ##拆分成单实体任务类型
110
  if j[1] =='DRUG_EFFICACY':
111
- output.append({'类型': j[1], '实体': j[-1]})
112
  output = json.dumps(output, ensure_ascii=False)
113
  data.append((input, output))
114
  return data
 
108
  for j in i['labels']:
109
  ##拆分成单实体任务类型
110
  if j[1] =='DRUG_EFFICACY':
111
+ output.append(j[-1])
112
  output = json.dumps(output, ensure_ascii=False)
113
  data.append((input, output))
114
  return data