cogdl.models.nn.pyg_dgcnn
¶
Module Contents¶
Classes¶
EdgeConv and DynamicGraph in paper `”Dynamic Graph CNN for Learning on |
-
class
cogdl.models.nn.pyg_dgcnn.
DGCNN
(in_feats, hidden_dim, out_feats, k=20, dropout=0.5)[source]¶ Bases:
cogdl.models.BaseModel
EdgeConv and DynamicGraph in paper “Dynamic Graph CNN for Learning on Point Clouds” <https://arxiv.org/pdf/1801.07829.pdf>__ .
- in_featsint
Size of each input sample.
- out_featsint
Size of each output sample.
- hidden_dimint
Dimension of hidden layer embedding.
- kint
Number of neareast neighbors.