CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-04-28 13:00:13 +00:00
Code Issues 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
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 data, metric in evals_log.items():
for metric_name, log in metric.items(): for metric_name, log in metric.items():
score = log[-1][0] if isinstance(log[-1], tuple) else log[-1] score = log[-1][0] if isinstance(log[-1], tuple) else log[-1]
if data == "train": key = self._get_key(data, metric_name)
self.writer.add_scalar("train_loss", score, epoch) self.writer.add_scalar(f"{key}_loss", score, epoch)
else:
self.writer.add_scalar("valid_loss", score, epoch)
return False return False