在Python中使用data_utils模块的namignizer_iterator()函数实现中文标题生成

import data_utils

# 读取训练数据

train_data = data_utils.load_data("train.txt")

# 构建字典

data_utils.create_vocab(train_data, "vocab.txt")

# 加载字典

vocab = data_utils.load_vocab("vocab.txt")

# 构建标题生成器

title_generator = data_utils.namignizer_iterator(train_data, vocab)

# 生成1000个标题

generated_titles = []

for _ in range(1000):

    title = next(title_generator)

    generated_titles.append(title)

print(generated_titles)