data wrappers
=============

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

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.ClusterWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.GraphSAGEDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.M3SDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.NetworkEmbeddingDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.FullBatchNodeClfDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.PPRGoDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.node_classification.SAGNDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:


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

.. autoclass:: cogdl.wrappers.data_wrapper.graph_classification.GraphClassificationDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.graph_classification.GraphEmbeddingDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.graph_classification.InfoGraphDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.graph_classification.PATCHY_SAN_DataWrapper
    :members:
    :undoc-members:
    :show-inheritance:


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

.. autoclass:: cogdl.wrappers.data_wrapper.pretraining.GCCDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:


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

.. autoclass:: cogdl.wrappers.data_wrapper.link_prediction.EmbeddingLinkPredictionDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.link_prediction.GNNKGLinkPredictionDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.link_prediction.GNNLinkPredictionDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:


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

.. autoclass:: cogdl.wrappers.data_wrapper.heterogeneous.HeterogeneousEmbeddingDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.heterogeneous.HeterogeneousGNNDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:

.. autoclass:: cogdl.wrappers.data_wrapper.heterogeneous.MultiplexEmbeddingDataWrapper
    :members:
    :undoc-members:
    :show-inheritance:



