JiaqiLee commited on
Commit
8fe687a
1 Parent(s): 7ab06e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = 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 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