pritmanvar commited on
Commit
381b231
·
verified ·
1 Parent(s): 478b653

resolved bug of dictionary

Browse files
Files changed (1) hide show
  1. NER/ner_inference.py +4 -4
NER/ner_inference.py CHANGED
@@ -36,9 +36,9 @@ def inference(sent):
36
  for label in res['entities']:
37
  sorted_labels = sorted(label['labels'], key=lambda k: k['confidence'])
38
  if sorted_labels[0]['value'] not in label_name:
39
- label_name[sorted_labels[0]['value']].append([label['text']])
40
  else:
41
- label_name[sorted_labels[0]['value']].append([label['text']])
42
 
43
  # RE part
44
  if label_name['GCNUM'] is None or len(label_name['GCNUM']) == 0:
@@ -47,8 +47,8 @@ def inference(sent):
47
  if result is not None and len(result) > 0:
48
  label_name['GCNUM'].append(result[0])
49
 
50
- label_name['GCNUM'] = [json.dumps(label_name['GCNUM'])]
51
- label_name['TRACK-ID'] = [json.dumps(label_name['TRACK-ID'])]
52
  #getting companie name
53
  #1. get name directly from sentence
54
  label_name['Company'] = get_company_name(sent.lower())
 
36
  for label in res['entities']:
37
  sorted_labels = sorted(label['labels'], key=lambda k: k['confidence'])
38
  if sorted_labels[0]['value'] not in label_name:
39
+ label_name[sorted_labels[0]['value']].append(label['text'])
40
  else:
41
+ label_name[sorted_labels[0]['value']].append(label['text'])
42
 
43
  # RE part
44
  if label_name['GCNUM'] is None or len(label_name['GCNUM']) == 0:
 
47
  if result is not None and len(result) > 0:
48
  label_name['GCNUM'].append(result[0])
49
 
50
+ label_name['GCNUM'] = label_name['GCNUM']
51
+ label_name['TRACK-ID'] = label_name['TRACK-ID']
52
  #getting companie name
53
  #1. get name directly from sentence
54
  label_name['Company'] = get_company_name(sent.lower())