(encoder_model, data)
| 56 | |
| 57 | |
| 58 | def test(encoder_model, data): |
| 59 | encoder_model.eval() |
| 60 | z, _, _ = encoder_model(data.x, data.edge_index, data.edge_attr) |
| 61 | split = get_split(num_samples=z.size()[0], train_ratio=0.1, test_ratio=0.8) |
| 62 | result = LREvaluator()(z, data.y, split) |
| 63 | return result |
| 64 | |
| 65 | |
| 66 | def main(): |
no test coverage detected