如何检查通知是否被提及

import re

def check_mention(notification, keywords):
    for keyword in keywords:
        pattern = r'\b' + re.escape(keyword) + r'\b'
        if re.search(pattern, notification, re.IGNORECASE):
            return True
    return False

# 示例使用
notification = "Hello @user1, your order has been shipped."
keywords = ["@user1", "@user2"]
if check_mention(notification, keywords):
    print("Notification mentions a user.")
else:
    print("No mention found.")

from transformers import pipeline

def check_mention_ml(notification, model_name="distilbert-base-uncased-finetuned-sst-2-english"):
    classifier = pipeline("text-classification", model=model_name)
    result = classifier(notification)
    return result[0]['label'] == 'POSITIVE'

# 示例使用
notification = "Hello @user1, your order has been shipped."
if check_mention_ml(notification):
    print("Notification mentions a user.")
else:
    print("No mention found.")