resolved bug of dictionary
Browse files- 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(
|
40 |
else:
|
41 |
-
label_name[sorted_labels[0]['value']].append(
|
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'] =
|
51 |
-
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())
|