-
Notifications
You must be signed in to change notification settings - Fork 1
/
tagging.py
35 lines (26 loc) · 1.06 KB
/
tagging.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
34
35
import nltk
from nltk.corpus import state_union
from nltk.tokenize import PunktSentenceTokenizer
# TRAINING: 2005 State of the Union address by Pres. George W. Bush
train_text = state_union.raw("2005-GWBush.txt")
# TESTING: 2006 State of the Union address by Pres. George W. Bush
sample_text = state_union.raw("2006-GWBush.txt")
# TRAIN the Punkt tokenizer
custom_sent_tokenizer = PunktSentenceTokenizer(train_text)
# tokenize
tokenized = custom_sent_tokenizer.tokenize(sample_text)
# iterate over text and tag POS per sentence
def process_content():
try:
for i in tokenized[:1]:
words = nltk.word_tokenize(i)
# convert unicode to str
str_words = []
for word in words:
str_words.append(str(word))
tagged = nltk.pos_tag(str_words)
print tagged
except Exception as e:
print str(e)
process_content()
# start_str = ["PRESIDENT GEORGE W. BUSH'S ADDRESS BEFORE A JOINT SESSION OF THE CONGRESS ON THE STATE OF THE UNION\n\nJanuary 31, 2006\n\nTHE PRESIDENT: Thank you all."]