chemprop.nn.utils
=================

.. py:module:: chemprop.nn.utils


Attributes
----------

.. autoapisummary::

   chemprop.nn.utils.logger


Classes
-------

.. autoapisummary::

   chemprop.nn.utils.Activation


Functions
---------

.. autoapisummary::

   chemprop.nn.utils.get_activation_function


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

.. py:data:: logger

.. py:class:: Activation

   Bases: :py:obj:`chemprop.utils.utils.EnumMapping`


   Enum where members are also (and must be) strings


   .. py:attribute:: RELU


   .. py:attribute:: LEAKYRELU


   .. py:attribute:: PRELU


   .. py:attribute:: TANH


   .. py:attribute:: ELU


.. py:function:: get_activation_function(activation)

   Gets an activation function module given the name of the activation.

   See :class:`~chemprop.v2.models.utils.Activation` for available activations.

   :param activation: The name of the activation function.
   :type activation: str | nn.Module | Activation

   :returns: The activation function module.
   :rtype: nn.Module