2020.10.1 发现最新的 pytorch 已经原生支持 one-hot 操作torch.nn.functional.one_hot()
pytorch 官方文档链接
只需如下一行代码:label_one_hot = torch.nn.functional.one_hot(labels, self.num_classes).float().to(self.device)
也可用如下的方法:
def encode_onehot(labels):
classes = set(labels)
classes_dict = {c: np.identity(len(classes))[i, :] for i, c in
enumerate(classes)}
labels_onehot = np.array(list(map(classes_dict.get, labels)),
dtype=np.int32)
return labels_onehot
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Pytorch下 label 的 one-hot 形式转换方法 - Python技术站