Я пытаюсь прочитать папку npz. В моей zip-папке (cora.npz) есть файлы npy в формате массива. Когда я пытаюсь загрузить разреженный граф из двоичного файла numpy, я получаю эту ошибку:
AttributeError Traceback (most recent call last)
<ipython-input-19-e58ac31009ee> in <module>
----> 1 _A_obs, _X_obs, _z_obs = utils.load_npz('data/cora.npz')
2 _A_obs = _A_obs + _A_obs.T
3 _A_obs[_A_obs > 1] = 1
4 lcc = utils.largest_connected_components(_A_obs)
5
~/Jupyter_Projects/NETTACK ALGORITHM/nettack/utils.py in load_npz(file_name)
28
29 if 'attr_data' in loader:
---> 30 attr_matrix = sp.csr_matrix((loader['attr_data'], loader['attr_indices'],
31 loader['attr_indptr']), shape=loader['attr_shape'])
32 else:
Мой код выглядит так:
if not file_name.endswith('.npz'):
file_name += '.npz'
with np.load(file_name) as loader:
loader = dict(loader)
adj_matrix = sp.csr_matrix((loader['adj_data'], loader['adj_indices'],
loader['adj_indptr']), shape=loader['adj_shape'])
if 'attr_data' in loader:
attr_matrix = sp.csr_matrix((loader['attr_data'], loader['attr_indices'],
loader['attr_indptr']), shape=loader['attr_shape'])
else:
attr_matrix = None
labels = loader.get('labels')
#print(labels)
return adj_matrix, attr_matrix, labels
Когда я пытаюсь прочитать мои файлы с np.load ('adj_indptr.npy', allow_pickle = True) считывающая часть успешно завершает работу и печатает числа внутри нее в формате массива.