1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-11 06:46:56 +08:00

Fix some API(for lb nn)

This commit is contained in:
Young
2021-04-07 03:31:50 +00:00
parent 8362780e22
commit 1dbb561744
5 changed files with 39 additions and 30 deletions

View File

@@ -145,6 +145,10 @@ class SignalRecord(RecordTemp):
del params["data_key"]
# The backend handler should be DataHandler
raw_label = self.dataset.prepare(**params)
except AttributeError:
# The data handler is initialize with `drop_raw=True`...
# So raw_label is not available
raw_label = None
self.recorder.save_objects(**{"label.pkl": raw_label})
self.dataset.__class__ = orig_cls