LCCFlccf的博客

最近在学习pytorch，有一个猫狗分类的kaggle比赛，预测结果要保存在csv文件中提交。其中格式要求第一列是id（纯数字），第二列是预测为狗的概率。 但我预测出来的id那一列里忘记tensor.item()操作了，以至于输出的不是纯数字，懒得在代码里改了，也抓住这个机会来学一学字符串的处理与csv的读取操作。这个是预测出来的图，很明显不符合赛方要求的格式需要将id这一列变成纯数字，如下...