adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add test for CNNPredictionModel

Browse Source
This commit is contained in:
robcaulk
2022-12-06 23:50:34 +01:00
parent 665eed3906
commit 389ab7e44b
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
freqtrade/freqai/base_models/BaseTensorFlowModel.py
View File

@@ -43,7 +43,6 @@ class BaseTensorFlowModel(IFreqaiModel):
start_time = time()
# filter the features requested by user in the configuration file and elegantly handle NaNs
features_filtered, labels_filtered = dk.filter_features(
unfiltered_df,
dk.training_features_list,

6
freqtrade/freqai/prediction_models/CNNPredictionModel.py
View File

@@ -14,9 +14,6 @@ from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
logger = logging.getLogger(__name__)
# tf.config.run_functions_eagerly(True)
# tf.data.experimental.enable_debug_mode()
class CNNPredictionModel(BaseTensorFlowModel):
"""
@@ -49,7 +46,8 @@ class CNNPredictionModel(BaseTensorFlowModel):
# we need to remove batch_size from the model_training_params because
# we dont want fit() to get the incorrect assignment (we use the WindowGenerator)
# to handle our batches.
self.model_training_parameters.pop('batch_size')
if 'batch_size' in self.model_training_parameters:
self.model_training_parameters.pop('batch_size')
input_dims = [BATCH_SIZE, self.CONV_WIDTH, n_features]
w1 = WindowGenerator(