Last Name Classification

🚀 姓氏分類模型

這是一個基於Transformer的分類器，可判斷所提供的姓氏是真實的（LABEL_1）還是虛假的（LABEL_0）。該模型有助於驗證聯繫表單提交信息、防止機器人錄入，或用於一般的姓名分類任務。

🚀 快速開始

本姓氏分類模型能有效識別姓氏的真實性，可通過以下步驟快速使用。

✨ 主要特性

• 基於Transformer架構，具備強大的特徵提取能力。
• 可用於驗證聯繫表單提交信息、防止機器人錄入等場景。
• 能對一般的姓名進行分類，判斷其真實性。

📦 安裝指南

克隆倉庫

git clone https://github.com/Vishodi/Last-Name-Classification.git

環境搭建

使用pip安裝所需的包：

pip install -r requirements.txt

💻 使用示例

基礎用法

from transformers import pipeline

# Replace with your model repository
model_dir = "vishodi/Last-Name-Classification"

# Load the model pipeline with authentication
classifier = pipeline(
    "text-classification",
    model=model_dir,
    tokenizer=model_dir,
)

# Test the model
test_names = ["musk", "zzzzzz", "uhyhu", "trump"]
for name in test_names:
    result = classifier(name)
    label = result[0]['label']
    score = result[0]['score']
    print(f"Name: {name} => Prediction: {label}, Score: {score:.4f}")

輸出結果：

Name: musk   => Prediction: LABEL_1, Score: 0.9167
Name: zzzzzz => Prediction: LABEL_0, Score: 0.9991
Name: uhyhu  => Prediction: LABEL_0, Score: 0.9944
Name: trump  => Prediction: LABEL_1, Score: 0.9998

📚 詳細文檔

項目結構

Last_Name_Prediction/
├── .gitattributes
├── README.md
├── config.json
├── model.safetensors
├── requirements.txt
├── special_tokens_map.json
├── tokenizer.json
├── tokenizer_config.json
└── vocab.txt

支持我們

📄 許可證

本項目採用MIT許可證。