Add scripts for Q

2021-03-11 13:07:08 +00:00
parent 58907a2387
commit 9f7eca0e58
5 changed files with 28 additions and 5 deletions
--- a/lib/trade_models/quant_transformer.py
+++ b/lib/trade_models/quant_transformer.py
@@ -17,7 +17,7 @@ import logging
 from qlib.utils import (
    unpack_archive_with_buffer,
    save_multiple_parts_file,
-    create_save_path,
+    get_or_create_path,
    drop_nan_by_y_index,
 )
 from qlib.log import get_module_logger, TimeInspector
@@ -176,7 +176,7 @@ class QuantTransformer(Model):
            _prepare_loader(test_dataset, False),
        )

-        save_path = create_save_path(save_path)
+        save_path = get_or_create_path(save_path)
        self.logger.info("Fit procedure for [{:}] with save path={:}".format(self.__class__.__name__, save_path))

        def _internal_test(ckp_epoch=None, results_dict=None):