|
112 | 112 | "source": [ |
113 | 113 | "## 2. Adding new models\n", |
114 | 114 | "\n", |
115 | | - "It is also possible to add entirely new models to AutoEmulate. AutoEmulate has a base `Emulator` class that handles most of the general functionality required for training and prediction. To implement a new emulator, one must simply subclass `Emulator` and implement two abstract methods: `_fit` and `_predict` as well any model specific functionality and initialisations.\n", |
| 115 | + "It is also possible to add entirely new models to AutoEmulate. AutoEmulate has a base `Emulator` class that handles most of the general functionality required for training and prediction. To implement a new emulator, one must simply subclass `Emulator` and implement the abstract methods (`_fit`, `_predict` and `is_multioutput`), `get_tune_params` to enable model tuning, as well any model specific functionality and initialisations.\n", |
116 | 116 | "\n", |
117 | 117 | "Since AutoEmulate supports a variety of models, there are additional `Emulator` subclasses that handle specific functionality for each model type:\n", |
118 | 118 | "- `PytorchBackend` for PyTorch models\n", |
119 | 119 | "- `SklearnBackend` for scikit-learn models\n", |
120 | | - "- `GaussianProcessExact` for exact Gaussian Process implementations\n", |
| 120 | + "- `GaussianProcess` for exact Gaussian Process implementations\n", |
121 | 121 | "- `Ensemble` for ensemble models\n", |
122 | 122 | "\n", |
123 | | - "Subclassing one of these directly has slightly different requirements. For example, when subclassing `PytorchBackend` or `GaussianProcessExact`, one must implement the `forward` method to define the model's forward pass. \n", |
| 123 | + "Subclassing one of these directly has slightly different requirements. For example, when subclassing `PytorchBackend` or `GaussianProcess`, one must implement the `forward` method to define the model's forward pass.\n", |
124 | 124 | "\n", |
125 | 125 | "There are also some static methods that should be implemented to provide metadata about the model, such as `is_multioutput` and `get_tune_params`.\n", |
126 | 126 | "\n", |
|
197 | 197 | ], |
198 | 198 | "metadata": { |
199 | 199 | "kernelspec": { |
200 | | - "display_name": "autoemulate", |
| 200 | + "display_name": ".venv", |
201 | 201 | "language": "python", |
202 | 202 | "name": "python3" |
203 | 203 | }, |
|
211 | 211 | "name": "python", |
212 | 212 | "nbconvert_exporter": "python", |
213 | 213 | "pygments_lexer": "ipython3", |
214 | | - "version": "3.12.7" |
| 214 | + "version": "3.12.11" |
215 | 215 | } |
216 | 216 | }, |
217 | 217 | "nbformat": 4, |
|
0 commit comments