CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-17 21:31:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

add config option for activating and deactivating tensorboard logger, ensure the various flavors are never activated simultaneously

Browse Source
This commit is contained in:
robcaulk
2023-05-14 14:08:00 +00:00
parent ab7a474ab6
commit 55a1a3afd6
11 changed files with 79 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
freqtrade/freqai/RL/BaseReinforcementLearningModel.py
View File

@@ -68,8 +68,11 @@ class BaseReinforcementLearningModel(IFreqaiModel):
self.unset_outlier_removal()
self.net_arch = self.rl_config.get('net_arch', [128, 128])
self.dd.model_type = import_str
self.tensorboard_callback: TensorboardCallback = \
TensorboardCallback(verbose=1, actions=BaseActions)
if self.activate_tensorboard:
self.tensorboard_callback: TensorboardCallback = \
TensorboardCallback(verbose=1, actions=BaseActions)
else:
self.tenorboard_callback = None
def unset_outlier_removal(self):
"""
@@ -156,7 +159,10 @@ class BaseReinforcementLearningModel(IFreqaiModel):
best_model_save_path=str(dk.data_path))
actions = self.train_env.get_actions()
self.tensorboard_callback = TensorboardCallback(verbose=1, actions=actions)
if self.activate_tensorboard:
self.tensorboard_callback = TensorboardCallback(verbose=1, actions=actions)
else:
self.tensorboard_callback = None # type: ignore
def pack_env_dict(self, pair: str) -> Dict[str, Any]:
"""