crackrammer/ShieldBERT-Base-Chinese-Sensitive

1"""2模型定义：基于 BertForSequenceClassification 的二分类模型3使用 HuggingFace 原生的 save_pretrained / from_pretrained 实现可靠保存/加载4"""5 6from transformers import BertForSequenceClassification7 8 9def create_model(model_name: str = "bert-base-chinese", num_labels: int = 2):10    """创建分类模型"""11    model = BertForSequenceClassification.from_pretrained(12        model_name,13        num_labels=num_labels,14    )15    return model16 17 18def load_model(model_path: str):19    """加载已训练的模型"""20    model = BertForSequenceClassification.from_pretrained(model_path)21    return model22