42 lines
866 B
ReStructuredText
42 lines
866 B
ReStructuredText
Nano HPO API
|
|
==================
|
|
|
|
Search Space
|
|
---------------------------
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Categorical
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Real
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Int
|
|
|
|
.. autoclass:: bigdl.nano.automl.hpo.space.Bool
|
|
|
|
|
|
HPO for Tensorflow
|
|
---------------------------
|
|
|
|
bigdl.nano.automl.tf.keras.Model
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. autoclass:: bigdl.nano.automl.tf.keras.Model.Model
|
|
:members: search, search_summary
|
|
|
|
|
|
bigdl.nano.automl.tf.keras.Sequential
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. autoclass:: bigdl.nano.automl.tf.keras.Sequential.Sequential
|
|
:members: search, search_summary
|
|
|
|
|
|
HPO for PyTorch
|
|
---------------------------
|
|
|
|
bigdl.nano.pytorch.Trainer
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. autoclass:: bigdl.nano.pytorch.Trainer
|
|
:members: search, search_summary
|
|
:undoc-members:
|
|
|