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)
