Skip to content

Commit

Permalink
Merge pull request #125 from FinAminToastCrunch/patch-2
Browse files Browse the repository at this point in the history
Updated so that this works with Windows
ChandlerBang authored Jan 5, 2023
2 parents 188caa1 + a1ce5b0 commit 1c0ef07
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion deeprobust/graph/data/dataset.py
Original file line number Diff line number Diff line change
@@ -9,6 +9,7 @@
from deeprobust.graph.utils import get_train_val_test, get_train_val_test_gcn
import zipfile
import json
import platform

class Dataset():
"""Dataset class contains four citation network datasets "cora", "cora-ml", "citeseer" and "pubmed",
@@ -61,7 +62,12 @@ def __init__(self, root, name, setting='nettack', seed=None, require_mask=False)
self.seed = seed
# self.url = 'https://raw.githubusercontent.com/danielzuegner/nettack/master/data/%s.npz' % self.name
self.url = 'https://raw.githubusercontent.com/danielzuegner/gnn-meta-attack/master/data/%s.npz' % self.name
self.root = osp.expanduser(osp.normpath(root))

if platform.system() == 'Windows':
root = root
else:
self.root = osp.expanduser(osp.normpath(root))

self.data_folder = osp.join(root, self.name)
self.data_filename = self.data_folder + '.npz'
self.require_mask = require_mask

0 comments on commit 1c0ef07

Please sign in to comment.