mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #9897 from freqtrade/fix/xgboosttensorboard

Browse Source 
fix: try plotting as much info in xgboost tensorboard as possible
This commit is contained in:
Matthias
2024-03-04 06:38:22 +01:00
committed by GitHub
parent 6bdb651573 a948796ef7
commit 426bc4c97b
2 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/prediction_models/XGBoostRegressor.py
+9 -2
View File
@@ -36,8 +36,15 @@ class XGBoostRegressor(BaseRegressionModel):
eval_set = None eval_set = None
eval_weights = None eval_weights = None
else: else:
eval_set = [(data_dictionary["test_features"], data_dictionary["test_labels"])] eval_set = [
eval_weights = [data_dictionary['test_weights']] (data_dictionary["test_features"],
data_dictionary["test_labels"]),
(X, y)
]
eval_weights = [
data_dictionary['test_weights'],
data_dictionary['train_weights']
]
sample_weight = data_dictionary["train_weights"] sample_weight = data_dictionary["train_weights"]
freqtrade/freqai/tensorboard/tensorboard.py
+4 -6
View File
@@ -43,13 +43,11 @@ class TensorBoardCallback(BaseTensorBoardCallback):
if not evals_log: if not evals_log:
return False return False
for data, metric in evals_log.items(): evals = ["validation", "train"]
for metric_name, log in metric.items(): for metric, eval in zip(evals_log.items(), evals):
for metric_name, log in metric[1].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": self.writer.add_scalar(f"{eval}-{metric_name}", score, epoch)
self.writer.add_scalar("train_loss", score, epoch)
else:
self.writer.add_scalar("valid_loss", score, epoch)
return False return False