(encoder_model, data)
| 70 | |
| 71 | |
| 72 | def test(encoder_model, data): |
| 73 | encoder_model.eval() |
| 74 | z1, z2, _, _, _, _ = encoder_model(data.x, data.edge_index) |
| 75 | z = z1 + z2 |
| 76 | split = get_split(num_samples=z.size()[0], train_ratio=0.1, test_ratio=0.8) |
| 77 | result = LREvaluator()(z, data.y, split) |
| 78 | return result |
| 79 | |
| 80 | |
| 81 | def main(): |
no test coverage detected