Views
No views yet
uethus| Tên model | Kiến trúc | Chức năng chính |
|---|---|---|
thanhtlx/text_classification_2 | RobertaForSequenceClassification | Chuẩn hóa tên cơ quan thành một trong 12 nhãn chuẩn |
str)smp: School of Medicine and Pharmacyueb: University of Economics and Businessussh: University of Social Sciences and Humanitieshus: University of Sciencehsb: Hanoi School of Businesslaw: School of Lawulis: University of Languages and International Studiesvju: Vietnam Japan Universityuet: University of Engineering and Technologyued: University of Educationis: International Schooliti: Information Technology Instituteother: Othersuethus1from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
2import tensorflow as tf
3
4# Tải tokenizer và mô hình
5tokenizer = AutoTokenizer.from_pretrained("thanhtlx/text_classification_2")
6model = TFAutoModelForSequenceClassification.from_pretrained("thanhtlx/text_classification_2")
7
8# Tên cơ quan cần chuẩn hóa
9text = "VNU University of Engineering and Technology"
10
11# Mã hóa văn bản
12inputs = tokenizer(text, return_tensors="tf", truncation=True, padding=True)
13
14# Dự đoán
15outputs = model(**inputs)
16logits = outputs.logits
17
18# Chuyển logits thành xác suất
19probs = tf.nn.softmax(logits, axis=1)
20
21# Lấy nhãn có xác suất cao nhất
22predicted_label = tf.argmax(probs, axis=1).numpy()[0]
23print(f"Nhãn dự đoán: {model.config.id2label[predicted_label]}")