XIn-9/trackBERT

BERT 大学生创新创业项目赛道适配预测模型

本模型基于 bert-base-chinese 微调，用于根据大学生创新创业项目名称或一句话简介预测适合的参赛赛道。

任务类型

本任务属于单标签多分类文本分类任务。模型输入一段中文项目文本，输出四个赛道类别之一：

• 产业赛道
• 高教主赛道
• 青年红色筑梦之旅赛道
• 职教赛道

阶段性评估结果

验证集样本数：1659

指标	数值
Accuracy	75.41%
Precision Micro	75.41%
Recall Micro	75.41%
F1 Micro	75.41%
Precision Macro	70.53%
Recall Macro	70.72%
F1 Macro	70.44%

各类别结果：

类别	Precision	Recall	F1
产业赛道	0.8200	0.8146	0.8173
高教主赛道	0.8146	0.8310	0.8227
青年红色筑梦之旅赛道	0.7102	0.7976	0.7514
职教赛道	0.4764	0.3855	0.4262

使用示例

import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification

model_id = "你的用户名/trackBERT"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)
model.eval()

text = "红色文化数字化传播与乡村振兴实践项目"
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)

with torch.no_grad():
    logits = model(**inputs).logits
    pred_id = torch.argmax(logits, dim=-1).item()

print(model.config.id2label[pred_id])

说明

当前模型主要基于项目名称训练，适合作为赛道适配预测的第一版 baseline。后续可通过加入项目简介、关键词、项目领域信息，以及优化类别不均衡问题进一步提升效果。