graph4nlp.prediction
=========================

.. automodule:: graph4nlp.prediction

Classification
--------------
.. automodule:: graph4nlp.prediction.classification

Graph Classification
++++++++++++++++++++

.. py:currentmodule:: graph4nlp.prediction.classification.graph_classification
.. automodule:: graph4nlp.prediction.classification.graph_classification

.. autoclass:: FeedForwardNN
    :members:

.. autoclass:: AvgPooling
    :members:

Knowledge Graph Completion
+++++++++++++++++++++++++++

.. py:currentmodule:: graph4nlp.prediction.classification.kg_completion
.. automodule:: graph4nlp.prediction.classification.kg_completion

.. autoclass:: ComplEx
    :members:

.. autoclass:: ComplExLayer
    :members:

.. autoclass:: DistMult
    :members:

.. autoclass:: DistMultLayer
    :members:

.. autoclass:: TransE
    :members:

.. autoclass:: TransELayer
    :members:

Link Prediction
++++++++++++++++

.. py:currentmodule:: graph4nlp.prediction.classification.link_prediction
.. automodule:: graph4nlp.prediction.classification.link_prediction

.. autoclass:: ConcatFeedForwardNN
    :members:

.. autoclass:: ConcatFeedForwardNNLayer
    :members:

.. autoclass:: ElementSum
    :members:

.. autoclass:: ElementSumLayer
    :members:

.. autoclass:: StackedElementProd
    :members:

.. autoclass:: StackedElementProdLayer
    :members:

Node Classification
+++++++++++++++++++

.. py:currentmodule:: graph4nlp.prediction.classification.node_classification
.. automodule:: graph4nlp.prediction.classification.node_classification

.. autoclass:: BiLSTMFeedForwardNN
    :members:

.. autoclass:: BiLSTMFeedForwardNNLayer
    :members:

.. autoclass:: FeedForwardNN
    :members:

.. autoclass:: FeedForwardNNLayer
    :members:

Generation
-----------

.. py:currentmodule:: graph4nlp.prediction.generation
.. automodule:: graph4nlp.prediction.generation

.. autoclass:: StdRNNDecoder
    :members:

.. autoclass:: StdTreeDecoder
    :members:

.. autoclass:: DecoderStrategy
    :members:

