SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Added the necessay import of module (#30804)

Browse Source
This commit is contained in:
Ankur Singh
2024-05-14 23:15:06 +05:30
committed by GitHub
parent ccdabc5642
commit c96aca3a8d
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/source/en/training.md
View File

@@ -186,6 +186,7 @@ so we can just convert that directly to a NumPy array without tokenization!
```py ```py
from transformers import AutoTokenizer from transformers import AutoTokenizer
import numpy as np
tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-cased") tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-cased")
tokenized_data = tokenizer(dataset["sentence"], return_tensors="np", padding=True) tokenized_data = tokenizer(dataset["sentence"], return_tensors="np", padding=True)