# API

## Preprocessing

```{eval-rst}
.. module:: cellpin.pp
.. currentmodule:: cellpin

.. autosummary::
    :toctree: generated

    pp.setup_data
    pp.load_sc_example
    pp.load_sp_example
```

## Tools

```{eval-rst}
.. module:: cellpin.tl
.. currentmodule:: cellpin

.. autosummary::
    :toctree: generated

    tl.label_transfer
```

## Plotting

```{eval-rst}
.. module:: cellpin.pl
.. currentmodule:: cellpin

.. autosummary::
    :toctree: generated

    pl.losses
```

## Models

```{eval-rst}
.. module:: cellpin.models
.. currentmodule:: cellpin

.. autosummary::
    :toctree: generated

    models.CellPin
```

### Model methods

```{eval-rst}
.. currentmodule:: cellpin.models

.. autosummary::
    :toctree: generated

    CellPin.fit
    CellPin.pretrain_model
    CellPin.train_model
    CellPin.impute
    CellPin.get_cell_embedding
```

## Dataset

```{eval-rst}
.. module:: cellpin.dataset
.. currentmodule:: cellpin

.. autosummary::
    :toctree: generated

    dataset.scAnnDataset
    dataset.stAnnDataset
```
