elliot.dataset package

Submodules

elliot.dataset.abstract_dataset module

class elliot.dataset.abstract_dataset.AbstractDataset(*args, **kwargs)[source]

Bases: object

abstract build_dict()[source]
abstract build_sparse(*args)[source]
abstract get_test(*args)[source]
required_attributes = ['config', 'args', 'kwargs', 'users', 'items', 'num_users', 'num_items', 'private_users', 'public_users', 'private_items', 'public_items', 'transactions', 'train_dict', 'i_train_dict', 'sp_i_train', 'test_dict']
class elliot.dataset.abstract_dataset.ForceRequiredAttributeDefinitionMeta[source]

Bases: type

check_required_attributes(class_object)[source]

elliot.dataset.dataset module

Module description:

class elliot.dataset.dataset.DataSet(*args, **kwargs)[source]

Bases: elliot.dataset.abstract_dataset.AbstractDataset

Load train and test dataset

build_dict(dataframe, users)[source]
build_sparse()[source]
build_sparse_ratings()[source]
dataframe_to_dict(data)[source]
get_test()[source]
get_validation()[source]
class elliot.dataset.dataset.DataSetLoader(config, *args, **kwargs)[source]

Bases: object

Load train and test dataset

check_timestamp(d: pandas.DataFrame) → pandas.DataFrame[source]
generate_dataobjects() → List[object][source]
generate_dataobjects_mock() → List[object][source]
read_splitting(folder_path)[source]

Module contents

Module description: