Skip to content

Commit

Permalink
Update README.md (#563)
Browse files Browse the repository at this point in the history
Wrong dataset used (`small_train_dataset`) to create `small_eval_dataset`.
  • Loading branch information
rbrisita authored Aug 20, 2024
1 parent df509b3 commit 51c0c67
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,7 @@ small_train_dataset = dataset["train"].shuffle(seed=42).select(range(1000))
small_eval_dataset = dataset["test"].shuffle(seed=42).select(range(300))

small_train_dataset = small_train_dataset.map(tokenize_function, batched=True)
small_eval_dataset = small_train_dataset.map(tokenize_function, batched=True)
small_eval_dataset = small_eval_dataset.map(tokenize_function, batched=True)

# download the model
model = AutoModelForSequenceClassification.from_pretrained(
Expand Down

0 comments on commit 51c0c67

Please sign in to comment.