yqy2001的博客

torch.max(input, dim)函数对dim维度求最大值，返回两个值，第一个是max value，第二个是max index。 分类问题计算accuracy： 根据预测结果列表prediction和label列表y计算: torch.eq函数比较两个tensor，返回一个bool类型的tensor，相同的地方为True，不同的地方为False；通过float()函数将其转换成0、1的float类型；mean()函数用来求平均值，即accuracy，计算出来的是一个tensor，item()用来取出

安装命令行： >>>pip install pytorch-crf 使用： from torchcrf import CRF

转载自：https://www.jb51.net/article/177575.htm，做简要记录 1、tensor与numpy数组： # tensor转numpy tensor.numpy() # numpy转tensor torch.from_numpy() 2、tensor与list： # tensor转list tensor.tolist() # list转tensor torch.tensor() 3、tensor类型转换 tensor = torch.Tensor(3, 5) # torch.

转载自：https://www.jianshu.com/p/6ba95579082c 只保存模型参数 # 保存 torch.save(model.state_dict(), '\parameter.pkl') # 加载 model = TheModelClass(...) model.load_state_dict(torch.load('\parameter.pkl')) 保存完整模型 # 保存 torch.save(model, '\model.pkl') # 加载 model = torch.load