cellpin.models.CellPin
======================

.. currentmodule:: cellpin.models

.. autoclass:: CellPin



Methods table
~~~~~~~~~~~~~

.. autosummary::

    ~CellPin.__init__
    ~CellPin.compute_losses
    ~CellPin.compute_pretrain_losses
    ~CellPin.configure_optimizers
    ~CellPin.embed_and_impute
    ~CellPin.fit
    ~CellPin.get_cell_embedding
    ~CellPin.impute
    ~CellPin.on_train_epoch_start
    ~CellPin.pretrain_model
    ~CellPin.save
    ~CellPin.set_stage_loss_weights
    ~CellPin.train_model
    ~CellPin.training_step
    ~CellPin.validation_step





Attributes
~~~~~~~~~~


.. autoattribute:: CellPin.training






Methods
~~~~~~~


.. automethod:: CellPin.compute_losses
.. automethod:: CellPin.compute_pretrain_losses
.. automethod:: CellPin.configure_optimizers
.. automethod:: CellPin.embed_and_impute
.. automethod:: CellPin.fit
.. automethod:: CellPin.get_cell_embedding
.. automethod:: CellPin.impute
.. automethod:: CellPin.on_train_epoch_start
.. automethod:: CellPin.pretrain_model
.. automethod:: CellPin.save
.. automethod:: CellPin.set_stage_loss_weights
.. automethod:: CellPin.train_model
.. automethod:: CellPin.training_step
.. automethod:: CellPin.validation_step