pytoda.datasets.dataframe_dataset module¶

KeyDataset for pandas DataFrames .

Summary¶

Classes:

DataFrameDataset

Dataset of rows from pandas.DataFrame

Reference¶

class DataFrameDataset(df)[source]¶

Bases: Generic[torch.utils.data.dataset.T_co]

Dataset of rows from pandas.DataFrame

get_key(index)[source]¶

Get key from integer index.

Return type: Hashable

get_index(key)[source]¶

Get index for first datum mapping to the given key.

Return type: int

get_item_from_key(key)[source]¶

Get item via key

Return type: array

keys()[source]¶

Iterator over index of dataframe.

Return type: Iterator