Advertisement

PyTorch与Visdom结合用于基于RNN的国籍预测

阅读量:

参考 Sean Robertson 提供的一个简明示例:https://github.com/spro/practical-pytorch

环境

系统:win10
cpu:i7-6700HQ
gpu:gtx965m
python : 3.6
pytorch :0.3

数据集

下载地址:https://download.pytorch.org/tutorial/data.zip

下载之后解压,放在项目根目录:

这里写图片描述

针对数据集

每个名字构建成 [字符数,1(batch数,因为不用故1),one-hot表示的字符位置]

复制代码
    # string.ascii_letters生成所有字母, string.digits 生成数字
    all_letters = string.ascii_letters+" .,;'"
    n_letters = len(all_letters)
    print(n_letters)
    # >>57

编码转换部分中存在一些

全部评论 (0)

还没有任何评论哟~