add classifier, improve model naming scheme

This commit is contained in:
Robert Caulk
2022-07-09 10:13:33 +02:00
parent ce8fbbf743
commit 07763d0d4f
12 changed files with 62 additions and 39 deletions

View File

@@ -19,15 +19,6 @@ class BaseRegressionModel(IFreqaiModel):
such as prediction_models/CatboostPredictionModel.py for guidance.
"""
def return_values(self, dataframe: DataFrame) -> DataFrame:
"""
User uses this function to add any additional return values to the dataframe.
e.g.
dataframe['volatility'] = dk.volatility_values
"""
return dataframe
def train(
self, unfiltered_dataframe: DataFrame, pair: str, dk: FreqaiDataKitchen
) -> Any: