9d0ed839dc9df4c9afc5966c1e3276c478a58d53,scripts/update_metadata_util.py,,load_task,#Any#,24

Before Change

    unique = np.unique(y_train)
    mapping = {unique_value: i for i, unique_value in enumerate(unique)}
    y_train = np.array([mapping[value] for value in y_train])
    y_test = np.array([mapping[value] for value in y_test])

    return X_train, y_train, X_test, y_test, cat

After Change

    del dataset
    cat = ["categorical" if c else "numerical" for c in cat]

    if isinstance(task, openml.tasks.OpenMLClassificationTask):
        task_type = "classification"
    elif isinstance(task, openml.tasks.OpenMLRegressionTask):
        task_type = "regression"
    else:
        raise ValueError("Unknown task type")

    return X_train, y_train, X_test, y_test, cat, task_type

Instances