model wrappers
==============

Node Classification
-------------------

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.DGIModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.GCNMixModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.GRACEModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.GrandModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.MVGRLModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.SelfAuxiliaryModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.GraphSAGEModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.UnsupGraphSAGEModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.M3SModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.NetworkEmbeddingModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.NodeClfModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.CorrectSmoothModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.PPRGoModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.node_classification.SAGNModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:


Graph Classification
--------------------

.. autoclass:: cogdl.wrappers.model_wrapper.graph_classification.GraphClassificationModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.graph_classification.GraphEmbeddingModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.graph_classification.InfoGraphModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:


Pretraining
-----------

.. autoclass:: cogdl.wrappers.model_wrapper.pretraining.GCCModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:


Link Prediction
---------------

.. autoclass:: cogdl.wrappers.model_wrapper.link_prediction.EmbeddingLinkPredictionModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.link_prediction.GNNKGLinkPredictionModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.link_prediction.GNNLinkPredictionModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:


Heterogeneous
-------------

.. autoclass:: cogdl.wrappers.model_wrapper.heterogeneous.HeterogeneousEmbeddingModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.heterogeneous.HeterogeneousGNNModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.heterogeneous.MultiplexEmbeddingModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:


Clustering
----------

.. autoclass:: cogdl.wrappers.model_wrapper.clustering.AGCModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.clustering.DAEGCModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.model_wrapper.clustering.GAEModelWrapper
    :members:
    :undoc-members:
    :show-inheritance:
