liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 00:36:55 +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) return np.mean(losses), np.mean(scores)
def fit( 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( 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: if df_train.empty or df_valid.empty:
raise ValueError("Empty data from dataset, please check your dataset config.") raise ValueError("Empty data from dataset, please check your dataset config.")