autogluon.core.TabularDataset

class autogluon.core.TabularDataset(data, **kwargs)

A dataset in tabular format (with rows = samples, columns = features/variables). This class returns a pd.DataFrame when initialized and all existing pandas methods can be applied to it. For full list of methods/attributes, see pandas Dataframe documentation: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.html

The purpose of this class is to provide an easy-to-use shorthand for loading a pandas DataFrame to use in AutoGluon.

Parameters:

data (str, pd.DataFrame, np.ndarray, Iterable, or dict) – If str, path to data file (CSV or Parquet format). If you already have your data in a pd.DataFrame, you can specify it here. In this case, the same DataFrame will be returned with no changes.

Examples

>>> import pandas as pd
>>> from autogluon.common import TabularDataset
>>> train_data = TabularDataset("https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv")
>>> train_data_pd = pd.read_csv("https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv")
>>> assert isinstance(train_data, pd.DataFrame)  # True
>>> assert train_data.equals(train_data_pd)  # True
>>> assert type(train_data) == type(train_data_pd)  # True

__init__()

Methods