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

cleaning and bug fixing

Browse Source
This commit is contained in:
robcaulk
2022-05-03 10:28:13 +02:00
parent fc837c4daa
commit b40f8f88ac
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
freqtrade/freqai/freqai_interface.py
View File

@@ -17,6 +17,7 @@ class IFreqaiModel(ABC):
User models should inherit from this class as shown in
templates/ExamplePredictionModel.py where the user overrides
train(), predict(), fit(), and make_labels().
Author: Robert Caulk, rob.caulk@gmail.com
"""
def __init__(self, config: Dict[str, Any]) -> None:
@@ -104,10 +105,10 @@ class IFreqaiModel(ABC):
return dataframe
def train(self, unfiltered_dataframe: DataFrame, metadata: dict) -> Tuple[DataFrame, DataFrame]:
def train(self, unfiltered_dataframe: DataFrame, metadata: dict) -> Any:
"""
Filter the training data and train a model to it. Train makes heavy use of the datahandler
for storing, saving, loading, and managed.
for storing, saving, loading, and analyzing the data.
:params:
:unfiltered_dataframe: Full dataframe for the current training period
:metadata: pair metadata from strategy.
@@ -115,7 +116,7 @@ class IFreqaiModel(ABC):
:model: Trained model which can be used to inference (self.predict)
"""
return unfiltered_dataframe, unfiltered_dataframe
return Any
def fit(self) -> Any:
"""