文件名称:pgn-summarization:指针生成器NN的实现
文件大小:11KB
文件格式:ZIP
更新时间:2024-05-18 18:40:59
Python
Pytorch指针生成器递归NN 模型 它是“模型进行的实现。 要求 python3 pytorch4 张量板 用法 准备 为了进行训练,有必要准备数据集的train.tsv和validation.tsv部分,并将其存储在公共文件夹中。 文档的每一行都有两部分:源和目标,由tab符分隔。 表格文件不需要标题。 训练 用于火车模型 $ python main.py 参数 可以使用--help参数获取参数的完整列表。 这是最重要的参数列表: --proceed继续学习最新模型 --dataset数据集的目录 --vocab_size词汇量,数据集中最常用单词的数量 --epochs纪元数 --hidden_size内部层的隐藏大小 --logdir日志目录 --model_name此名称用于保存学习检查点 训练停止时,模型状态将保存到--model_name文件。 如果--proce
【文件预览】:
pgn-summarization-master
----data.py(9KB)
----utils.py(3KB)
----main.py(5KB)
----model.py(14KB)
----LICENSE(1KB)
----README.md(1KB)
----.gitignore(1KB)