Update README.md
Browse files
README.md
CHANGED
@@ -23,12 +23,12 @@ from transformers import BertForSequenceClassification, BertTokenizer, TextClass
|
|
23 |
model_path = "JiaqiLee/bert-agnews"
|
24 |
tokenizer = BertTokenizer.from_pretrained(model_path)
|
25 |
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=4)
|
26 |
-
pipeline =
|
27 |
print(pipeline("Google scores first-day bump of 18 (USATODAY.com): USATODAY.com - Even a big first-day jump in shares of Google (GOOG) couldn't quiet debate over whether the Internet search engine's contentious auction was a hit or a flop."))
|
28 |
```
|
29 |
|
30 |
## Training data
|
31 |
-
The training data comes HuggingFace [AGNews dataset](https://huggingface.co/datasets/ag_news). We use 90% of the `train.csv` data to train the model and the remaining 10% for evaluation.
|
32 |
|
33 |
## Evaluation results
|
34 |
|
|
|
23 |
model_path = "JiaqiLee/bert-agnews"
|
24 |
tokenizer = BertTokenizer.from_pretrained(model_path)
|
25 |
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=4)
|
26 |
+
pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
|
27 |
print(pipeline("Google scores first-day bump of 18 (USATODAY.com): USATODAY.com - Even a big first-day jump in shares of Google (GOOG) couldn't quiet debate over whether the Internet search engine's contentious auction was a hit or a flop."))
|
28 |
```
|
29 |
|
30 |
## Training data
|
31 |
+
The training data comes from HuggingFace [AGNews dataset](https://huggingface.co/datasets/ag_news). We use 90% of the `train.csv` data to train the model and the remaining 10% for evaluation.
|
32 |
|
33 |
## Evaluation results
|
34 |
|