WeightBasisο
- class dgl.nn.pytorch.utils.WeightBasis(shape, num_bases, num_outputs)[source]ο
Bases:
Module
Basis decomposition from Modeling Relational Data with Graph Convolutional Networks
It can be described as below:
Each weight output
is essentially a linear combination of basis transformations with coefficients .If is useful as a form of regularization on a large parameter matrix. Thus, the number of weight outputs is usually larger than the number of bases.
- Parameters: