classify

from textblob import Word
import re

def parse(text):
    pattern = '''Synset\(\'(.*)\.n'''
    return re.findall(pattern,text)[0]

def classify(word):
    try:
        word = Word(word)
    except:
        return 'no category'
    try:
        type_of_word = word.synsets[0].hypernyms()[0]
        return parse(str(type_of_word))
    except:
        return 'no category'

words = ["Cat","Castle","King","Lion","Swan","Tavern"]
for word in words:
    print word, "->", classify(word)
    
# Cat -> feline
# Castle -> mansion
# King -> sovereign
# Lion -> big_cat
# Swan -> aquatic_bird
# Tavern -> building