CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-02-06 14:20:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

try limiting tb_logger to pytorch only (XGBoost still gets its callback)

Browse Source
This commit is contained in:
robcaulk
2023-05-14 12:03:15 +00:00
parent 8a9b2fc16f
commit ab7a474ab6
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
freqtrade/freqai/freqai_interface.py
View File

@@ -635,9 +635,11 @@ class IFreqaiModel(ABC):
dk.find_features(unfiltered_dataframe)
dk.find_labels(unfiltered_dataframe)
self.tb_logger = TBLogger(dk.data_path)
if self.dd.model_type == "pytorch":
self.tb_logger = TBLogger(dk.data_path)
model = self.train(unfiltered_dataframe, pair, dk)
self.tb_logger.close()
if self.dd.model_type == "pytorch":
self.tb_logger.close()
self.dd.pair_dict[pair]["trained_timestamp"] = trained_timestamp
dk.set_new_model_names(pair, trained_timestamp)