We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 14171a5 commit 5c6933cCopy full SHA for 5c6933c
docs/parameters.rst
@@ -1,7 +1,14 @@
1
Parameters
2
==========
3
4
-This page provides the API reference of :mod:`torchensemble`.
+This page provides the API reference of :mod:`torchensemble`. Below is a list of functions supported by all ensembles.
5
+
6
+* :meth:`fit`: Training stage of the ensemble
7
+* :meth:`evaluate`: Evaluating stage of the ensemble
8
+* :meth:`predict`: Return the predictions of the ensemble
9
+* :meth:`forward`: Data forward process of the ensemble
10
+* :meth:`set_optimizer`: Set the parameter optimizer for training the ensemble
11
+* :meth:`set_scheduler`: Set the learning rate scheduler for training the ensemble
12
13
Fusion
14
------
0 commit comments