Skip to content
Permalink
Browse files

fixed test

  • Loading branch information...
rusty1s committed Mar 27, 2019
1 parent a8aece6 commit f07ef3212796da98f26ed4d28bf0839ba3f49ed4
Showing with 2 additions and 2 deletions.
  1. +1 −0 .gitignore
  2. +1 −2 test/nn/models/test_autoencoder.py
@@ -7,6 +7,7 @@ alpha/
.eggs/
*.egg-info/
.coverage
.coverage.*

!torch_geometric/data/
!test/data/
@@ -32,8 +32,7 @@ def test_gae():
assert data.train_neg_adj_mask.sum().item() == (11**2 - 11) / 2 - 4 - 6 - 5

z = torch.randn(11, 16)
loss = model.recon_loss(z, data.train_pos_edge_index,
data.train_neg_adj_mask)
loss = model.recon_loss(z, data.train_pos_edge_index)
assert loss.item() > 0

auc, ap = model.test(z, data.val_pos_edge_index, data.val_neg_edge_index)

0 comments on commit f07ef32

Please sign in to comment.
You can’t perform that action at this time.