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

Fix CI (#1529)

Browse Source
This commit is contained in:
you-n-g
2023-05-31 08:39:52 +08:00
committed by GitHub
parent efffb2819a
commit 0e9ac9dce7
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
qlib/contrib/model/pytorch_sandwich.py
View File

@@ -300,10 +300,15 @@ class Sandwich(Model):
return np.mean(losses), np.mean(scores)
def fit(
self, dataset: DatasetH, evals_result=dict(), save_path=None,
self,
dataset: DatasetH,
evals_result=dict(),
save_path=None,
):
df_train, df_valid, df_test = dataset.prepare(
["train", "valid", "test"], col_set=["feature", "label"], data_key=DataHandlerLP.DK_L,
["train", "valid", "test"],
col_set=["feature", "label"],
data_key=DataHandlerLP.DK_L,
)
if df_train.empty or df_valid.empty:
raise ValueError("Empty data from dataset, please check your dataset config.")