def download(self):
print("Downloading {} dataset.".format(self.name))
url = self.url.format(self.name)
download_file(url, self.path, self.name + ".zip")
# Datasets are zipped in a folder: unpack them
parent = self.path
subfolder = osp.join(self.path, self.name)
for filename in os.listdir(subfolder):
shutil.move(osp.join(subfolder, filename),
osp.join(parent, filename))
os.rmdir(subfolder)
x, adj, y, mask_tr, mask_va, mask_te = preprocess_data(
self.path, self.name)
# Save pre-processed data
npz_file = osp.join(self.path, self.name) + ".npz"
adj = adj.tocoo()
np.savez(
npz_file,
x=x,
adj_data=adj.data,
adj_row=adj.row,
adj_col=adj.col,
adj_shape=adj.shape,
y=y,
mask_tr=mask_tr,
mask_va=mask_va,
mask_te=mask_te,
)
def download(self):
print('Downloading {} dataset{}.'
.format(self.name, ' (clean)' if self.clean else ''))
url = '{}/{}.zip'.format(self.url_clean if self.clean else self.url, self.name)
download_file(url, self.path, self.name + '.zip')
# Datasets are zipped in a folder: unpack them
parent = self.path
subfolder = osp.join(self.path, self.name)
for filename in os.listdir(subfolder):
shutil.move(osp.join(subfolder, filename), osp.join(parent, filename))
os.rmdir(subfolder)
def download(self):
print("Downloading ModelNet{} dataset.".format(self.name))
url = self.url[self.name]
download_file(url, self.path, "ModelNet" + self.name + ".zip")
# Datasets are zipped in a folder: unpack them
parent = self.path
subfolder = osp.join(self.path, "ModelNet" + self.name)
for filename in os.listdir(subfolder):
shutil.move(osp.join(subfolder, filename),
osp.join(parent, filename))
os.rmdir(subfolder)
shutil.rmtree(osp.join(self.path, "__MACOSX"), ignore_errors=True)
def download(self):
print("Downloading ModelNet{} dataset.".format(self.name))
url = self.url[self.name]
download_file(url, self.path, "ModelNet" + self.name + ".zip")
shutil.rmtree(osp.join(self.path, "__MACOSX"), ignore_errors=True)
def download(self):
print('Downloading ModelNet{} dataset.'.format(self.name))
url = self.url[self.name]
download_file(url, self.path, 'ModelNet' + self.name + '.zip')
shutil.rmtree(osp.join(self.path, '__MACOSX'), ignore_errors=True)