(path)
| 10 | |
| 11 | |
| 12 | def get_data(path): |
| 13 | with open(path + "/adjacency_matrices.pkl", "rb") as f: |
| 14 | adj_matrices = pickle.load(f) |
| 15 | |
| 16 | with open(path + "/feature_matrices.pkl", "rb") as f: |
| 17 | feature_matrices = pickle.load(f) |
| 18 | |
| 19 | labels = np.load(path + "/labels.npy") |
| 20 | |
| 21 | return adj_matrices, feature_matrices, labels |
| 22 | |
| 23 | def create_random_split(path): |
| 24 | adj_matrices, feature_matrices, labels = get_data(path) |
no test coverage detected