def dataset_recorder_env(request): return Environment( train_dataset=small_toy_dataset, holdout_dataset=getattr(request, "param", None), metrics=["accuracy_score"], target_column="t", cv_params=dict(n_splits=4, shuffle=True, random_state=32), experiment_callbacks=[dataset_recorder()], )
def boston_env(): return Environment( train_dataset=boston_head, holdout_dataset=holdout_last_row, target_column="DIS", metrics=["r2_score", "median_absolute_error"], cv_type="RepeatedKFold", cv_params=dict(n_repeats=2, n_splits=3, random_state=1), experiment_callbacks=[dataset_recorder()], )