liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 07:46:53 +08:00
Code Activity

Fix test_all_pipeline.py

Browse Source
This commit is contained in:
zhupr
2020-09-29 19:16:18 +08:00
committed by you-n-g
parent e22e98ddc9
commit 2c4b31b274
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/get_data.py
View File

@@ -53,23 +53,25 @@ class GetData:
for _file in tqdm(zp.namelist()): for _file in tqdm(zp.namelist()):
zp.extract(_file, str(target_dir.resolve())) zp.extract(_file, str(target_dir.resolve()))
def qlib_data_cn(self, target_dir="~/.qlib/qlib_data/cn_data", version="latest"): def qlib_data_cn(self, name="qlib_data_cn", target_dir="~/.qlib/qlib_data/cn_data", version="latest"):
"""download cn qlib data from remote """download cn qlib data from remote
Parameters Parameters
---------- ----------
target_dir: str target_dir: str
data save directory data save directory
name: str
dataset name, value from [qlib_data_cn, qlib_data_cn_simple], by default qlib_data_cn
version: str version: str
data version, value from [v0, v1, ..., latest], by default latest data version, value from [v0, v1, ..., latest], by default latest
Examples Examples
--------- ---------
python get_data.py qlib_data_cn --target_dir ~/.qlib/qlib_data/cn_data --version v1 python get_data.py qlib_data_cn --target_dir ~/.qlib/qlib_data/cn_data --version latest
------- -------
""" """
file_name = f"qlib_data_cn_{version}.zip" file_name = f"{name}_{version}.zip"
self._download_data(file_name, target_dir) self._download_data(file_name, target_dir)
def csv_data_cn(self, target_dir="~/.qlib/csv_data/cn_data"): def csv_data_cn(self, target_dir="~/.qlib/csv_data/cn_data"):

10
tests/test_all_pipeline.py
View File

@@ -141,14 +141,14 @@ class TestAllFlow(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls) -> None: def setUpClass(cls) -> None:
# use default data # use default data
provier_uri = "~/.qlib/qlib_data/cn_data" # target_dir provider_uri = "~/.qlib/qlib_data/cn_data" # target_dir
if not exists_qlib_data(provier_uri): if not exists_qlib_data(provider_uri):
print(f"Qlib data is not found in {provier_uri}") print(f"Qlib data is not found in {provider_uri}")
sys.path.append(str(Path(__file__).resolve().parent.parent.joinpath("scripts"))) sys.path.append(str(Path(__file__).resolve().parent.parent.joinpath("scripts")))
from get_data import GetData from get_data import GetData
GetData().qlib_data_cn(provier_uri) GetData().qlib_data_cn(provider_uri)
qlib.init(provier_uri=provier_uri, region=REG_CN) qlib.init(provider_uri=provider_uri, region=REG_CN)
def test_0_train(self): def test_0_train(self):
TestAllFlow.PRED_SCORE, model_pearsonr = train() TestAllFlow.PRED_SCORE, model_pearsonr = train()