Training function
-----------------------

.. autofunction:: celltypist.train