Назад к вопросам
Лига Цифровой Экономики
Middle
2
# Необходимо реализовать генератор, который принимает два аргумента: имя файла filename и кол-во слов batch_size. # Генератор должен читать заданный файл и за каждую итерацию вызова возвращать batch_size слов.
Компании, где спрашивали
Ответ от нейросети
sobes.tech AI
python def word_batch_generator(filename, batch_size): with open(filename, 'r', encoding='utf-8') as file: batch = [] for line in file: words = line.split() for word in words: batch.append(word) if len(batch) == batch_size: yield batch batch = [] if batch: yield batch
Этот генератор читает файл построчно, разбивает строки на слова и собирает их в списки размером batch_size. При достижении нужного размера возвращает текущий батч. В конце, если остались слова, возвращает их тоже.