feat: add SKLearnRandomForestClassifier

This commit is contained in:
robcaulk
2023-12-18 12:56:04 +01:00
parent 99359a7f85
commit 2f2467256c
3 changed files with 85 additions and 0 deletions

View File

@@ -176,6 +176,7 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model, s
'CatboostClassifier',
'XGBoostClassifier',
'XGBoostRFClassifier',
'SKLearnRandomForestClassifier',
'PyTorchMLPClassifier',
])
def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):

View File

@@ -1770,6 +1770,7 @@ def test_api_freqaimodels(botclient, tmp_path, mocker):
{'name': 'LightGBMRegressorMultiTarget'},
{'name': 'ReinforcementLearner'},
{'name': 'ReinforcementLearner_multiproc'},
{'name': 'SKlearnRandomForestClassifier'},
{'name': 'XGBoostClassifier'},
{'name': 'XGBoostRFClassifier'},
{'name': 'XGBoostRFRegressor'},
@@ -1788,6 +1789,7 @@ def test_api_freqaimodels(botclient, tmp_path, mocker):
'LightGBMRegressorMultiTarget',
'ReinforcementLearner',
'ReinforcementLearner_multiproc',
'SKlearnRandomForestClassifier',
'XGBoostClassifier',
'XGBoostRFClassifier',
'XGBoostRFRegressor',