29 lines
675 B
Python
29 lines
675 B
Python
|
from itertools import islice
|
||
|
|
import regex as re
|
||
|
|
import sys
|
||
|
|
from torchtext.vocab import build_vocab_from_iterator
|
||
|
|
import lzma
|
||
|
|
import utils
|
||
|
|
import torch
|
||
|
|
def get_word_lines_from_file(file_name):
|
||
|
|
counter=0
|
||
|
|
with lzma.open(file_name, 'r') as fh:
|
||
|
|
for line in fh:
|
||
|
|
counter+=1
|
||
|
|
# if counter == 4000:
|
||
|
|
# break
|
||
|
|
line = line.decode("utf-8")
|
||
|
|
yield utils.get_words_from_line(line)
|
||
|
|
|
||
|
|
|
||
|
|
vocab_size = utils.vocab_size
|
||
|
|
|
||
|
|
vocab = build_vocab_from_iterator(
|
||
|
|
get_word_lines_from_file('train/in.tsv.xz'),
|
||
|
|
max_tokens = vocab_size,
|
||
|
|
specials = ['<unk>', '<empty>'])
|
||
|
|
|
||
|
|
|
||
|
|
import pickle
|
||
|
|
with open("vocab.pickle", 'wb') as handle:
|
||
|
|
pickle.dump(vocab, handle)
|