A property prediction layer maps a graph’s learned latent representation to a predicted property value. In molecular graph generation, it adds property control alongside graph reconstruction. section 5.4.3
The implementation follows a short pipeline:
- Encode the graph with a heterogeneous GCN, preserving distinctions between bond types. section 5.4.3
Molecular property prediction
A VGAE predicts QED by defining property_prediction_layer = nn.Linear(latent_dim, 1), then producing qed_pred from the sampled latent representation.
