Fork me on GitHub

优雅代码收集

#标签读取后由one-hot编码转化为数值
labels=torch.from_numpy(data['label'].todense())
labels=labels.nonzero()[:,1]

#GCN导入字符串列表labels转化为one-hot编码
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)

  

#标签读取后由one-hot编码转化为数值
labels=torch.from_numpy(data['label'].todense())
labels=labels.nonzero()[:,1]
#GCN导入字符串列表labels转化为one-hot编码
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)

 

posted @ 2022-09-15 10:44  天淇  阅读(49)  评论(0)    收藏  举报
Live2D