df = pd.DataFrame(
np.random.randint(0, 10, (10, 3)),
index=np.random.choice('a b c'.split(), 10),
columns='x y z'.split()
)
df.head()Dataset Base and Meta
Base Classes for Datasets
DataModules
BaseDataFrameDataModule
BaseDataFrameDataModule (df:pandas.core.frame.DataFrame=<factory>)
set_dataset
set_dataset (DatasetClass:Any=<class 'litds.abc.dfds.base.BaseDataFrameDataset'>)
Example
@set_dataset(BaseDataFrameDataset)
class DFModuleTest(BaseDataFrameDataModule):
passdm = DFModuleTest(df=df)dm.setup()DataFrameDataModuleTest()
ds = dm.make_dataset()ds.loc['a']| x | y | z | |
|---|---|---|---|
| a | 0 | 9 | 0 |
| a | 1 | 3 | 9 |