Skip to content

Commit

Permalink
fixed sample params
Browse files Browse the repository at this point in the history
  • Loading branch information
Wollents committed Nov 10, 2024
1 parent 9ac0167 commit 7d81689
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions pygod/test/test_card.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,8 @@ class TestCARD(unittest.TestCase):
def setUp(self):
self.roc_floor = 0.55

self.train_data = torch.load(os.path.join('./train_graph.pt'))
self.test_data = torch.load(os.path.join('./test_graph.pt'))
self.train_data = torch.load(os.path.join('pygod/test/train_graph.pt'))
self.test_data = torch.load(os.path.join('pygod/test/test_graph.pt'))

def test_full(self):
detector = CARD(epoch=10, num_layers=2, hid_dim=8)
Expand Down Expand Up @@ -66,13 +66,13 @@ def test_sample(self):
num_layers=2,
dropout=0.5,
weight_decay=0.001,
act=None,
act=torch.nn.functional.relu,
backbone=GCN,
contamination=0.2,
lr=0.001,
epoch=10,
batch_size=16,
num_neigh=1,
num_neigh=4,
verbose=3,
save_emb=True,
act_first=True)
Expand Down

0 comments on commit 7d81689

Please sign in to comment.