import torch
import torch.nn as nn
import torch.optim as optim
import csv
from torch.utils.data import Dataset
from torch.utils.data import DataLoader
import numpy as np
import matplotlib.pyplot as plt
import time
from torch.nn.utils.rnn import pack_padded_sequence # https://pytorch.org/docs/master/generated/torch.nn.utils.rnn.pack_padded_sequence.html
import math
# 1:数据集
# 超参数
HIDDEN_SIZE = 100 # 隐藏层
BATCH_SIZE = 256
N_LAYER = 2 # RNN的层数
N_EPOCHS = 100 # train的轮数
N_CHARS = 128 # 这个就是要构造的字典的长度
USE_GPU = False
class NameDataset(Dataset): # 这个是自己写的数据集的类,就那3个函数
def __init__(self, is_train_set=True):
filename = "E:\\PyTorch\\PyTorch深度学习实践\\names_train.csv" if is_train_set else "E:\\PyTorch\\PyTorch深度学习实践\\names_test.csv"
with open(filename, "rt") as f: # 因为这个文件不是很大,所以在初始化的时候就全读进来了
reader = csv.reader(f)
rows = list(reader)
self.names = [row[0] for row in rows]
self
GRU实战双向
最新推荐文章于 2022-10-10 13:56:26 发布