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

add XGBoostRF models to freqai test interface

Browse Source
This commit is contained in:
silur 2022-10-11 13:17:21 +02:00
parent 2ad086dd7a
commit 30a45bb597
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/freqai/test_freqai_interface.py
View File

@ -30,6 +30,7 @@ def is_mac() -> bool:
@pytest.mark.parametrize('model', [ @pytest.mark.parametrize('model', [
'LightGBMRegressor', 'LightGBMRegressor',
'XGBoostRegressor', 'XGBoostRegressor',
'XGBoostRFRegressor',
'CatboostRegressor', 'CatboostRegressor',
]) ])
def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model): def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model):
@ -111,6 +112,7 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model):
'LightGBMClassifier', 'LightGBMClassifier',
'CatboostClassifier', 'CatboostClassifier',
'XGBoostClassifier', 'XGBoostClassifier',
'XGBoostRFClassifier',
]) ])
def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model): def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
if is_arm() and model == 'CatboostClassifier': if is_arm() and model == 'CatboostClassifier':