forked from swapnil-saxena/address-parser
-
Notifications
You must be signed in to change notification settings - Fork 0
/
predict.py
33 lines (25 loc) · 1.23 KB
/
predict.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import spacy
nlp=spacy.load("output\models\model-best")
address_list=["130 W BOSE ST STE 100, PARK RIDGE, IL, 60068, USA",
"8311 MCDONALD RD, HOUSTON, TX, 77053-4821, USA",
"PO Box 317, 4100 Hwy 20 E Ste 403, NICEVILLE, FL, 32578-5037, USA",
"C/O Elon Musk Innovations Inc, 1548 E Florida Avenue, Suite 209, TAMPA, FL, 33613, USA",
"Seven Edgeway Plaza, C/O Mac Dermott Inc, OAKBROOK TERRACE, IL, 60181, USA"]
# Checking predictions for the NER model
for address in address_list:
doc=nlp(address)
ent_list=[(ent.text, ent.label_) for ent in doc.ents]
print("Address string -> "+address)
print("Parsed address -> "+str(ent_list))
print("******")
address="C/o John Doe LLC, 111 8th Avenue Ste 1301, Tulsa, Oklahoma, 74136–1922, USA"
doc=nlp(address)
ent_list=[(ent.text, ent.label_) for ent in doc.ents]
print("Address string -> "+address)
print("Parsed address -> "+str(ent_list))
# Loading Entity Ruler coupled NER model and checking prediction
nlp=spacy.load("output\models_er\model-best")
doc=nlp(address)
ent_list=[(ent.text, ent.label_) for ent in doc.ents]
print("Address string -> "+address)
print("Parsed address -> "+str(ent_list))