Merge pull request #7390 from aemr3/add-xgboostclassifier

Add XGBoostClassifier
2022-09-12 23:38:04 +02:00
parent fac8f19554 330d7068ab
commit 26441820a9
2 changed files with 86 additions and 0 deletions
--- a/tests/freqai/test_freqai_interface.py
+++ b/tests/freqai/test_freqai_interface.py
@@ -99,6 +99,7 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model):
@pytest.mark.parametrize('model', [
    'LightGBMClassifier',
    'CatboostClassifier',
+    'XGBoostClassifier',
    ])
 def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
    if is_arm() and model == 'CatboostClassifier':