Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,7 @@ from transformers import BertTokenizer, BertForSequenceClassification
|
|
| 11 |
import torch
|
| 12 |
|
| 13 |
# 加载已训练的模型和分词器
|
| 14 |
-
model_path = 'qixun/
|
| 15 |
tokenizer = BertTokenizer.from_pretrained(model_path)
|
| 16 |
model = BertForSequenceClassification.from_pretrained(model_path)
|
| 17 |
|
|
@@ -41,4 +41,7 @@ predicted_label, probabilities = classify_text(text)
|
|
| 41 |
print(f"预测标签: {predicted_label}")
|
| 42 |
print(f"概率分布: {probabilities}")
|
| 43 |
|
| 44 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
import torch
|
| 12 |
|
| 13 |
# 加载已训练的模型和分词器
|
| 14 |
+
model_path = 'qixun/tangsong_poem_classify'
|
| 15 |
tokenizer = BertTokenizer.from_pretrained(model_path)
|
| 16 |
model = BertForSequenceClassification.from_pretrained(model_path)
|
| 17 |
|
|
|
|
| 41 |
print(f"预测标签: {predicted_label}")
|
| 42 |
print(f"概率分布: {probabilities}")
|
| 43 |
|
| 44 |
+
```
|
| 45 |
+
|
| 46 |
+
label_0代表唐诗风格
|
| 47 |
+
label_1代表宋诗风格
|