JiaqiLee commited on
Commit
28cb550
1 Parent(s): d256144

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/imdb-finetuned-bert-base-uncased"
24
  tokenizer = BertTokenizer.from_pretrained(model_path)
25
  model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
26
- pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
27
  print(pipeline("The movie depicted well the psychological battles that Harry Vardon fought within himself, from his childhood trauma of being evicted to his own inability to break that glass ceiling that prevents him from being accepted as an equal in English golf society."))
28
  ```
29
 
30
  ## Training data
31
- The training data comes HuggingFace [IMDB dataset](https://huggingface.co/datasets/imdb). 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/imdb-finetuned-bert-base-uncased"
24
  tokenizer = BertTokenizer.from_pretrained(model_path)
25
  model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
26
+ pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
27
  print(pipeline("The movie depicted well the psychological battles that Harry Vardon fought within himself, from his childhood trauma of being evicted to his own inability to break that glass ceiling that prevents him from being accepted as an equal in English golf society."))
28
  ```
29
 
30
  ## Training data
31
+ The training data comes from HuggingFace [IMDB dataset](https://huggingface.co/datasets/imdb). We use 90% of the `train.csv` data to train the model and the remaining 10% for evaluation.
32
 
33
  ## Evaluation results
34