.. _train:

Interface
=======================


Train
-----

`unimol_tools.train.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/train.py>`_ trains a Uni-Mol model.

.. automodule:: unimol_tools.train
   :members:


Predict
------------

`unimol_tools.predictor.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/predictor.py>`_ predict through a Uni-Mol model.

.. automodule:: unimol_tools.predict
   :members:


Uni-Mol representation
------------------------

`unimol_tools.predictor.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/predictor.py>`_ get the Uni-Mol representation.

.. automodule:: unimol_tools.predictor
   :members: