随笔分类 - pytorch
pytorch的topk()函数
摘要：pytorch.topk()用于返回Tensor中的前k个元素以及元素对应的索引值。例： import torch item=torch.IntTensor([1,2,4,7,3,2]) value,indices=torch.topk(item,3) print("value:",value) p
阅读全文
使用pytorch框架实现使用FM模型在movielen数据集上的电影评分预测(rendle的工作)
摘要：一、FM介绍 （1）实验的主要任务：使用FM在movielen数据集上进行电影评分预测任务（rendle的工作，经典的特征选择） （2）参考论文：Factorization Machines （3）部署环境：python37 + pytorch1.3 （4）数据集：Movielen的small数据集
阅读全文
使用pytorch框架实现使用MF模型在movielen数据集上的电影评分预测
摘要：一、MF介绍 （1）实验的主要任务：使用MF模型在数据集合上的评分预测（movielens，随机80%训练数据，20%测试数据，随机构造 Koren的经典模型） （2）参考论文：MATRIX FACTORIZATION TECHNIQUES FOR RECOMMENDER SYSTEMS 简单模型：
阅读全文
pytorch中使用torch.utils.data.TensorDataset时报错TypeError: 'int' object is not callable，同时在代码中并没有与TensorDataset重名的函数的解决办法。
摘要：使用TensorDataset函数的代码为： train_dataset = Data.TensorDataset(x_train,y_train) 执行之后发现报错： TypeError: 'int' object is not callable。但是检查代码发现并没有与TensorDataset
阅读全文
TypeError: Image data of dtype object cannot be converted to float
摘要：源码如下： 程序运行时一直在标红的代码处报错，找了好久才发现自己犯了十分愚蠢的错误:路径写错了（同时路径中含有了中文），虽然是路径错了，但一直报的是另外一个错误，所以也很难发现这个问题。 所以当你也出现了同样的错误，可以去检查一下路径中是否含有中文名或者路径写错了。解决方案仅供参考，如果有更好的方案
阅读全文
