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

add LightGBMClassifierMultiTarget. add test

Browse Source
This commit is contained in:
robcaulk
2022-11-11 17:45:53 +01:00
parent 63458a6130
commit 66514e84e4
2 changed files with 65 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/freqai/test_freqai_interface.py
View File

@@ -79,8 +79,7 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model):
('LightGBMRegressorMultiTarget', "freqai_test_multimodel_strat"),
('XGBoostRegressorMultiTarget', "freqai_test_multimodel_strat"),
('CatboostRegressorMultiTarget', "freqai_test_multimodel_strat"),
# ('LightGBMClassifierMultiTarget', "freqai_test_multimodel_classifier_strat"),
# ('XGBoostClassifierMultiTarget', "freqai_test_multimodel_classifier_strat"),
('LightGBMClassifierMultiTarget', "freqai_test_multimodel_classifier_strat"),
('CatboostClassifierMultiTarget', "freqai_test_multimodel_classifier_strat")
])
def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model, strat):