(labels)
| 4 | |
| 5 | |
| 6 | def encode_onehot(labels): |
| 7 | classes = set(labels) |
| 8 | classes_dict = {c: np.identity(len(classes))[i, :] for i, c in |
| 9 | enumerate(classes)} |
| 10 | labels_onehot = np.array(list(map(classes_dict.get, labels)), |
| 11 | dtype=np.int32) |
| 12 | return labels_onehot |
| 13 | |
| 14 | |
| 15 | def load_data(path="../data/cora/", dataset="cora"): |