-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathAirline2.py
More file actions
9 lines (6 loc) · 724 Bytes
/
Airline2.py
File metadata and controls
9 lines (6 loc) · 724 Bytes
1
2
3
4
5
6
7
8
9
from textblob.classifiers import NaiveBayesClassifier
train = [('Air India did a poor job of queue management both times.', 'staff service'),("The 'cleaning' by flight attendants involved regularly spraying air freshener in the lavatories.", 'staff'),('The food tasted decent.', 'food'),('Flew Air India direct from New York to Delhi round trip.', 'route'),('Colombo to Moscow via Delhi.', 'route'),('Flew Birmingham to Delhi with Air India.', 'route'),('Without toilet, food or anything!', 'food'),('Cabin crew announcements included a sincere apology for the delay.', 'cabin flown')]
cl = NaiveBayesClassifier(train)
tests = ['Food is good.', 'Colombo to Moscow via Delhi.']
for c in tests:
print c,'\t',cl.classify(c)