def load_puma32h(return_X_y=False, encode=True, verbose=False, onehot_threshold=10): data, meta = read_arff_data('data/regression/puma32h/puma32h.dat') db = pd.DataFrame(data) db.columns = list(db.columns[:-1]) + ['target'] return construct_return_set(db, "puma32h", return_X_y, encode, citation='keel', name="puma32h", verbose=verbose, problem_type='regression')
def load_zoo(return_X_y=False, encode=True, verbose=False, onehot_threshold=10): data, meta = read_arff_data('data/classification/zoo/zoo.dat') db = pd.DataFrame(data) db.columns = list(db.columns[:-1]) + ['target'] return construct_return_set(db, "zoo", return_X_y, encode, citation='keel', name="zoo", verbose=verbose, problem_type='classification', onehot_threshold=onehot_threshold)