lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: try plotting as much info in xgboost tensorboard as possible

Browse Source
This commit is contained in:
robcaulk 2024-03-03 12:23:18 +01:00
parent f8cc2a6e74
commit b8a1089592
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
freqtrade/freqai/tensorboard/tensorboard.py
View File

@ -46,10 +46,8 @@ class TensorBoardCallback(BaseTensorBoardCallback):
for data, metric in evals_log.items():
for metric_name, log in metric.items():
score = log[-1][0] if isinstance(log[-1], tuple) else log[-1]
if data == "train":
self.writer.add_scalar("train_loss", score, epoch)
else:
self.writer.add_scalar("valid_loss", score, epoch)
key = self._get_key(data, metric_name)
self.writer.add_scalar(f"{key}_loss", score, epoch)
return False