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

add test for CNNPredictionModel

Browse Source
This commit is contained in:
robcaulk
2022-12-06 23:50:34 +01:00
parent 665eed3906
commit 389ab7e44b
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/freqai/test_freqai_interface.py
View File

@@ -34,7 +34,8 @@ def is_mac() -> bool:
('CatboostRegressor', False, False, False),
('ReinforcementLearner', False, True, False),
('ReinforcementLearner_multiproc', False, False, False),
('ReinforcementLearner_test_4ac', False, False, False)
('ReinforcementLearner_test_4ac', False, False, False),
('CNNPredictionModel', False, False, False)
])
def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca, dbscan, float32):
if is_arm() and model == 'CatboostRegressor':
@@ -71,6 +72,10 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca,
if 'test_4ac' in model:
freqai_conf["freqaimodel_path"] = str(Path(__file__).parents[1] / "freqai" / "test_models")
if 'CNNPredictionModel' in model:
freqai_conf['freqai']['model_training_parameters'].pop('n_estimators')
model_save_ext = 'h5'
strategy = get_patched_freqai_strategy(mocker, freqai_conf)
exchange = get_patched_exchange(mocker, freqai_conf)
strategy.dp = DataProvider(freqai_conf, exchange)