windlx/url-classifier-lora

URL Page Type Classifier (LoRA)

基于 Qwen2.5-1.5B + LoRA 微调的URL类型分类模型，用于判断URL是列表页还是详情页。

模型信息

项目	详情
基础模型	Qwen/Qwen2.5-1.5B
微调方法	LoRA (r=16, alpha=32)
可训练参数	~18M (1.18%)

性能测试

测试集	样本数	准确率
训练数据	100	100%
随机生成URL	1000	100%

使用方法

from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel

# 加载基础模型
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B", device_map="auto", trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-1.5B", trust_remote_code=True)

# 加载LoRA
model = PeftModel.from_pretrained(base_model, "windlx/url-classifier-lora")
model.eval()

# 推理
url = "https://example.com/product/12345"
# ... (推理代码)

相关链接

• Merged模型: https://huggingface.co/windlx/url-classifier-model
• GitHub: https://github.com/xiuxiu/url-classifier