Update Classifier docstrings

This commit is contained in:
Matthias 2022-10-10 12:13:41 +00:00
parent d3b2b2972e
commit 60de192d47
3 changed files with 6 additions and 9 deletions

View File

@ -20,8 +20,7 @@ class CatboostClassifier(BaseClassifierModel):
def fit(self, data_dictionary: Dict, dk: FreqaiDataKitchen, **kwargs) -> Any:
"""
User sets up the training and test data to fit their desired model here
:params:
:data_dictionary: the dictionary constructed by DataHandler to hold
:param data_dictionary: the dictionary constructed by DataHandler to hold
all the training and test data/labels.
"""

View File

@ -20,8 +20,7 @@ class LightGBMClassifier(BaseClassifierModel):
def fit(self, data_dictionary: Dict, dk: FreqaiDataKitchen, **kwargs) -> Any:
"""
User sets up the training and test data to fit their desired model here
:params:
:data_dictionary: the dictionary constructed by DataHandler to hold
:param data_dictionary: the dictionary constructed by DataHandler to hold
all the training and test data/labels.
"""

View File

@ -26,8 +26,7 @@ class XGBoostClassifier(BaseClassifierModel):
def fit(self, data_dictionary: Dict, dk: FreqaiDataKitchen, **kwargs) -> Any:
"""
User sets up the training and test data to fit their desired model here
:params:
:data_dictionary: the dictionary constructed by DataHandler to hold
:param data_dictionary: the dictionary constructed by DataHandler to hold
all the training and test data/labels.
"""