rafiki.model
====================================================================

.. contents:: Table of Contents

Core Classes
--------------------------------------------------------------------

.. autoclass:: rafiki.model.BaseModel
    :members:

.. autoclass:: rafiki.model.BaseKnob
    :members:


.. _`knob-types`:

Knob Classes
--------------------------------------------------------------------

.. autoclass:: rafiki.model.CategoricalKnob
    :members:

.. autoclass:: rafiki.model.IntegerKnob
    :members:

.. autoclass:: rafiki.model.FloatKnob
    :members:

.. autoclass:: rafiki.model.FixedKnob
    :members:


Utility Classes & Methods
--------------------------------------------------------------------

.. automethod:: rafiki.model.test_model_class

.. autoclass:: rafiki.model.ModelLogger
    :members:

.. autoclass:: rafiki.model.ModelDatasetUtils
    :members:

.. autoclass:: rafiki.model.ImageFilesDataset

.. autoclass:: rafiki.model.CorpusDataset