:py:mod:`n2j.models.gnn`
========================

.. py:module:: n2j.models.gnn

.. autoapi-nested-parse::

   Various GNN models



Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   n2j.models.gnn.GCNNet
   n2j.models.gnn.GATNet
   n2j.models.gnn.SageNet
   n2j.models.gnn.GravNet




.. py:class:: GCNNet(in_channels, out_channels, hidden_channels=256, n_layers=3, dropout=0.0, kwargs={})

   Bases: :py:obj:`torch.nn.Module`

   .. py:method:: forward(data)



.. py:class:: GATNet(in_channels, out_channels, hidden_channels=256, kwargs={}, n_layers=3, dropout=0.0)

   Bases: :py:obj:`torch.nn.Module`

   .. py:method:: forward(data)



.. py:class:: SageNet(in_channels, out_channels, hidden_channels=256, n_layers=3, dropout=0.0, kwargs={})

   Bases: :py:obj:`torch.nn.Module`

   .. py:method:: forward(data)



.. py:class:: GravNet(in_channels, out_channels, hidden_channels=256, n_layers=3, dropout=0.0, kwargs={})

   Bases: :py:obj:`torch.nn.Module`

   .. py:method:: forward(data)