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

Update record to support time series dataset.

Browse Source
This commit is contained in:
lwwang1995
2020-12-06 18:41:04 +08:00
committed by you-n-g
parent a88697151a
commit a7c6aea386
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
qlib/workflow/record_temp.py
View File

@@ -141,6 +141,12 @@ class SignalRecord(RecordTemp):
del params["data_key"] del params["data_key"]
# The backend handler should be DataHandler # The backend handler should be DataHandler
raw_label = DatasetH.prepare(**params) raw_label = DatasetH.prepare(**params)
if not isinstance(raw_label, pd.DataFrame):
index = raw_label.get_index()
raw_label = raw_label.data.loc[index]
raw_label = raw_label.iloc[:,-1:]
self.recorder.save_objects(**{"label.pkl": raw_label}) self.recorder.save_objects(**{"label.pkl": raw_label})
def list(self): def list(self):