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

Merge pull request #10711 from freqtrade/fix/pytorch-scaling
Some checks failed
Build Documentation / Deploy Docs through mike (push) Has been cancelled
Details

Browse Source 
fix: Update BasePyTorchRegressor.py
This commit is contained in:
Matthias 2024-09-26 18:53:14 +02:00 committed by GitHub
commit 91d9c9b4d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
freqtrade/freqai/base_models/BasePyTorchRegressor.py
View File

@ -86,9 +86,6 @@ class BasePyTorchRegressor(BasePyTorchModel):
dk.feature_pipeline = self.define_data_pipeline(threads=dk.thread_count)
dk.label_pipeline = self.define_label_pipeline(threads=dk.thread_count)
dd["train_labels"], _, _ = dk.label_pipeline.fit_transform(dd["train_labels"])
dd["test_labels"], _, _ = dk.label_pipeline.transform(dd["test_labels"])
(dd["train_features"], dd["train_labels"], dd["train_weights"]) = (
dk.feature_pipeline.fit_transform(
dd["train_features"], dd["train_labels"], dd["train_weights"]

2
freqtrade/freqai/prediction_models/PyTorchTransformerRegressor.py
View File

@ -141,7 +141,7 @@ class PyTorchTransformerRegressor(BasePyTorchRegressor):
pred_df = pd.DataFrame(yb.detach().numpy(), columns=dk.label_list)
pred_df, _, _ = dk.label_pipeline.inverse_transform(pred_df)
if self.freqai_info.get("DI_threshold", 0) > 0:
if self.ft_params.get("DI_threshold", 0) > 0:
dk.DI_values = dk.feature_pipeline["di"].di_values
else:
dk.DI_values = np.zeros(outliers.shape[0])