Source code for pysubgroup.datasets
from io import StringIO
import pandas as pd
import pkg_resources
from scipy.io import arff
[docs]def get_credit_data():
s_io = StringIO(
str(pkg_resources.resource_string("pysubgroup", "data/credit-g.arff"), "utf-8")
)
return pd.DataFrame(arff.loadarff(s_io)[0])
[docs]def get_titanic_data():
s_io = StringIO(
str(pkg_resources.resource_string("pysubgroup", "data/titanic.csv"), "utf-8")
)
return pd.read_csv(s_io, sep="\t", header=[0])