From eabd321281a3dd112ed0dbcd325a1b28b9b5625a Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Thu, 23 Mar 2023 15:59:57 +0200 Subject: [PATCH] small docs change --- docs/freqai-configuration.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/freqai-configuration.md b/docs/freqai-configuration.md index 63d7c571c..92cff06ff 100644 --- a/docs/freqai-configuration.md +++ b/docs/freqai-configuration.md @@ -283,7 +283,7 @@ In addition, the trainer is responsible for the following: Like all freqai models, PyTorch models inherit `IFreqaiModel`. `IFreqaiModel` declares three abstract methods: `train`, `fit`, and `predict`. we implement these methods in three levels of hierarchy. From top to bottom: -1. `BasePyTorchModel` - Implements the `train` method. all `BasePyTorch*` inherit it. responsible for general data preparation (e.g., data normalization) and calling the `fit` method. Sets `device _type` attribute used by children classes. Sets `model_type` attribute used by the parent class. +1. `BasePyTorchModel` - Implements the `train` method. all `BasePyTorch*` inherit it. responsible for general data preparation (e.g., data normalization) and calling the `fit` method. Sets `device` attribute used by children classes. Sets `model_type` attribute used by the parent class. 2. `BasePyTorch*` - Implements the `predict` method. Here, the `*` represents a group of algorithms, such as classifiers or regressors. responsible for data preprocessing, predicting, and postprocessing if needed. 3. `PyTorch*Classifier` / `PyTorch*Regressor` - implements the `fit` method. responsible for the main train flaw, where we initialize the trainer and model objects.