elliot.dataset package

Subpackages

• elliot.dataset.dataloader package
  • Submodules
  • elliot.dataset.dataloader.knowledge_aware_chains module
  • elliot.dataset.dataloader.visual_dataloader module
  • Module contents
• elliot.dataset.samplers package
  • Submodules
  • elliot.dataset.samplers.custom_pointwise_sparse_sampler module
  • elliot.dataset.samplers.custom_sampler module
  • elliot.dataset.samplers.custom_sparse_sampler module
  • elliot.dataset.samplers.pairwise_sampler module
  • elliot.dataset.samplers.pipeline_sampler module
  • elliot.dataset.samplers.pointwise_cfgan_sampler module
  • elliot.dataset.samplers.pointwise_pos_neg_ratings_sampler module
  • elliot.dataset.samplers.pointwise_pos_neg_ratio_ratings_sampler module
  • elliot.dataset.samplers.pointwise_pos_neg_sampler module
  • elliot.dataset.samplers.pointwise_wide_and_deep_sampler module
  • elliot.dataset.samplers.sparse_sampler module
  • Module contents

Submodules

elliot.dataset.abstract_dataset module

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

Bases: object

abstract build_dict()

abstract build_sparse(*args)

abstract get_test(*args)

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

Bases: type

check_required_attributes(class_object)

elliot.dataset.dataset module

Module description:

class elliot.dataset.dataset.DataSet(*args, **kwargs)

Bases: elliot.dataset.abstract_dataset.AbstractDataset

Load train and test dataset

align_with_training(train, side_information_data)

Alignment with training

build_dict(dataframe, users)

build_sparse()

build_sparse_ratings()

dataframe_to_dict(data)

get_test()

get_validation()

to_bool_sparse(test_dict)

class elliot.dataset.dataset.DataSetLoader(config, *args, **kwargs)

Bases: elliot.dataset.modular_loaders.loader_coordinator_mixin.LoaderCoordinator

Load train and test dataset

check_timestamp(d: pandas.DataFrame) → pandas.DataFrame

generate_dataobjects() → List[object]

generate_dataobjects_mock() → List[object]

read_splitting(folder_path, column_names)

Module contents

Module description: