This commit is contained in:
you-n-g 2023-05-31 08:39:52 +08:00 коммит произвёл GitHub
Родитель efffb2819a
Коммит 0e9ac9dce7
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 7 добавлений и 2 удалений

Просмотреть файл

@ -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.")