Kategoriser tekst udfra dictionary med kategorier og tilhørende nøgleord

text_classify.py

# Nøgleord jeg ønsker at søge efter og tagge med key
keywords = {
    'belysning': ['lampe', 'lys', 'mørke'],
    'affald': ['skrald', 'affald', 'skraldespand']
}

# Ekempel på tre beskeder som skal klassificeres
msg1 = 'Der ikke lys i lamperne i Ballerup'
msg2 = 'De flyder med affald i gågaden'
msg3 = 'Der ikke lys og den skraldespand'
msg4 = 'Det sner i Afrika'

# Funktion til at bestemme om en besked indeholder nøgleord
def classify(msg, keywords):
    
    categories = []
    
    for cat, words in keywords.items():
        if any(word in msg.split(' ') for word in words):
            categories.append( cat )
        else:
            None
    
    return categories

print(classify(msg1, keywords))
print(classify(msg2, keywords))
print(classify(msg3, keywords))
print(classify(msg4, keywords))