reduce mlp number of parameters for testing

2023-03-13 20:09:12 +02:00
parent 9c8c30b0e8
commit 918889a2bd
2 changed files with 2 additions and 4 deletions
--- a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
+++ b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
@@ -35,7 +35,6 @@ class PyTorchClassifierMultiTarget(BasePyTorchModel):

        super().__init__(**kwargs)
        model_training_params = self.freqai_info.get("model_training_parameters", {})
-        self.n_hidden: int = model_training_params.get("n_hidden", 1024)
        self.max_iters: int = model_training_params.get("max_iters", 100)
        self.batch_size: int = model_training_params.get("batch_size", 64)
        self.learning_rate: float = model_training_params.get("learning_rate", 3e-4)