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

create more flexible whitelist, avoid duplicating whitelist features into corr_pairlist, update docs

Browse Source
This commit is contained in:
robcaulk
2022-05-09 17:01:49 +02:00
parent c151277705
commit 81f227cd9e
6 changed files with 119 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
freqtrade/templates/ExamplePredictionModel.py
View File

@@ -53,7 +53,7 @@ class ExamplePredictionModel(IFreqaiModel):
logger.info("--------------------Starting training--------------------")
# create the full feature list based on user config info
self.dh.training_features_list = self.dh.build_feature_list(self.config)
self.dh.training_features_list = self.dh.build_feature_list(self.config, metadata)
unfiltered_labels = self.make_labels(unfiltered_dataframe)
# filter the features requested by user in the configuration file and elegantly handle NaNs
@@ -114,7 +114,8 @@ class ExamplePredictionModel(IFreqaiModel):
return model
def predict(self, unfiltered_dataframe: DataFrame) -> Tuple[DataFrame, DataFrame]:
def predict(self, unfiltered_dataframe: DataFrame, metadata: dict) -> Tuple[DataFrame,
DataFrame]:
"""
Filter the prediction features data and predict with it.
:param: unfiltered_dataframe: Full dataframe for the current backtest period.
@@ -126,7 +127,7 @@ class ExamplePredictionModel(IFreqaiModel):
# logger.info("--------------------Starting prediction--------------------")
original_feature_list = self.dh.build_feature_list(self.config)
original_feature_list = self.dh.build_feature_list(self.config, metadata)
filtered_dataframe, _ = self.dh.filter_features(
unfiltered_dataframe, original_feature_list, training_filter=False
)