消息推送系统

import pika

import json

from sklearn.feature_extraction.text import CountVectorizer

from sklearn.naive_bayes import MultinomialNB

# 模拟训练数据

X = ["紧急", "正常", "紧急", "正常"]

y = [1, 0, 1, 0]

vectorizer = CountVectorizer()

X_vec = vectorizer.fit_transform(X)

model = MultinomialNB()

model.fit(X_vec, y)

def callback(ch, method, properties, body):

message = body.decode('utf-8')

print(f"收到消息: {message}")

X_test = vectorizer.transform([message])

prediction = model.predict(X_test)

if prediction[0] == 1:

print("警告：这是紧急消息！")

connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))

channel = connection.channel()

channel.queue_declare(queue='ai_messages')

channel.basic_consume(callback, queue='ai_messages', no_ack=True)

print('等待消息...')

channel.start_consuming()