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

Very small change to one of the function parameters (#32548)

Browse Source 
Very small change to one of the parameters

np.random.randint second parameter is not included in the possible options. Therefore, we want the upper range to be 2, so that we have some 1 labels in our classification as well.
This commit is contained in:
Ali Salamatian
2024-08-27 09:29:05 -07:00
committed by GitHub
parent 6101d934a1
commit 27903de7ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/en/model_memory_anatomy.md
View File

@@ -42,7 +42,7 @@ In total, we get 512 sequences each with length 512 and store them in a [`~datas
>>> seq_len, dataset_size = 512, 512 >>> seq_len, dataset_size = 512, 512
>>> dummy_data = { >>> dummy_data = {
... "input_ids": np.random.randint(100, 30000, (dataset_size, seq_len)), ... "input_ids": np.random.randint(100, 30000, (dataset_size, seq_len)),
... "labels": np.random.randint(0, 1, (dataset_size)), ... "labels": np.random.randint(0, 2, (dataset_size)),
... } ... }
>>> ds = Dataset.from_dict(dummy_data) >>> ds = Dataset.from_dict(dummy_data)
>>> ds.set_format("pt") >>> ds.set_format("pt")