消息推送系统

from sklearn.naive_bayes import MultinomialNB

from sklearn.feature_extraction.text import CountVectorizer

# 示例数据

messages = ["你中奖了！", "会议时间调整", "请确认订单"]

labels = ["spam", "normal", "normal"]

vectorizer = CountVectorizer()

X = vectorizer.fit_transform(messages)

model = MultinomialNB()

model.fit(X, labels)

# 测试新消息

new_message = ["恭喜你获得大奖！"]

企业级单点登录解决方案

new_X = vectorizer.transform(new_message)

print(model.predict(new_X)) # 输出: ['spam']