Skip to content

Commit 4e2ffa5

Browse files
fix: replace _build_vocab by train
for clarity and consistency
1 parent 93a6e80 commit 4e2ffa5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

torchTextClassifiers/tokenizers/ngram.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ def __init__(
274274
self.eos_token_id = 2
275275

276276
if training_text is not None:
277-
self._build_vocab(training_text)
277+
self.train(training_text)
278278
else:
279279
self.word_to_id = {}
280280
self.id_to_word = {}

0 commit comments

Comments
 (0)