adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

revert to using model_training_parameters

Browse Source
This commit is contained in:
Yinon Polak 2023-03-13 00:35:51 +02:00
parent b6096efadd
commit d7ea750823
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
View File

@ -34,13 +34,15 @@ class PyTorchClassifierMultiTarget(BasePyTorchModel):
""" """
super().__init__(**kwargs) super().__init__(**kwargs)
trainer_kwargs = self.freqai_info.get("trainer_kwargs", {}) model_training_params = self.freqai_info.get("model_training_parameters", {})
self.n_hidden: int = trainer_kwargs.get("n_hidden", 1024) self.n_hidden: int = model_training_params.get("n_hidden", 1024)
self.max_iters: int = trainer_kwargs.get("max_iters", 100) self.max_iters: int = model_training_params.get("max_iters", 100)
self.batch_size: int = trainer_kwargs.get("batch_size", 64) self.batch_size: int = model_training_params.get("batch_size", 64)
self.learning_rate: float = trainer_kwargs.get("learning_rate", 3e-4) self.learning_rate: float = model_training_params.get("learning_rate", 3e-4)
self.max_n_eval_batches: Optional[int] = trainer_kwargs.get("max_n_eval_batches", None) self.max_n_eval_batches: Optional[int] = model_training_params.get(
self.model_kwargs: Dict = trainer_kwargs.get("model_kwargs", {}) "max_n_eval_batches", None
)
self.model_kwargs: Dict = model_training_params.get("model_kwargs", {})
self.class_name_to_index = None self.class_name_to_index = None
self.index_to_class_name = None self.index_to_class_name = None