Ejemplos de dataset_recorder en Python

Ejemplo n.º 1

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()],
    )

Ejemplo n.º 2

Archivo: test_both_stages_transform.py Proyecto: zwcdp/hyperparameter_hunter

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()],
    )