Source code for plsxml.data.loader
import os
from ..plsxml import PLSXML
__all__ = ['data_names', 'data_path', 'load_data']
DATA_FOLDER = os.path.dirname(os.path.abspath(__file__))
METADATA = dict(
galloping=dict(
file='galloping.xml',
tables=None
),
galloping_zip=dict(
file='galloping.zip',
tables=None
)
)
[docs]def data_names():
"""
Returns a list of built-in dataset names.
"""
return sorted(METADATA.keys())
[docs]def data_path(name):
"""
Returns the path for the built-in dataset.
Parameters
----------
name : str
The name of the dataset.
"""
return os.path.join(DATA_FOLDER, METADATA[name]['file'])
[docs]def load_data(name):
"""
Returns a PLSXML object for the built-in dataset.
Parameters
----------
name : str
The name of the dataset.
"""
path = data_path(name)
return PLSXML(path, METADATA[name]['tables'])