优雅代码收集
#标签读取后由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)