Sobes.tech
Middle

# Պետք է իրականացնել գեներատոր, որը ընդունում է երկու արգումենտ՝ ֆայլի անունը filename և բառերի քանակը batch_size: # Գեներատորը պետք է կարդա նշված ֆայլը և յուրաքանչյուր կանչի ժամանակ վերադարձնի batch_size բառ։

sobes.tech AI

Պատասխան 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. При достижении нужного размера возвращает текущий батч. В конце, если остались слова, возвращает их тоже.