recbole.dataΒΆ
- recbole.data.dataloader
- recbole.data.dataloader.abstract_dataloader
AbstractDataLoader
AbstractDataLoader._dataset
AbstractDataLoader.shuffle
AbstractDataLoader.pr
AbstractDataLoader.step
AbstractDataLoader._batch_size
AbstractDataLoader.batch_size
AbstractDataLoader.collate_fn()
AbstractDataLoader.dataset
AbstractDataLoader.drop_last
AbstractDataLoader.num_workers
AbstractDataLoader.pin_memory
AbstractDataLoader.pin_memory_device
AbstractDataLoader.prefetch_factor
AbstractDataLoader.sampler
AbstractDataLoader.set_batch_size()
AbstractDataLoader.timeout
AbstractDataLoader.update_config()
NegSampleDataLoader
NegSampleDataLoader.batch_size
NegSampleDataLoader.dataset
NegSampleDataLoader.drop_last
NegSampleDataLoader.get_model()
NegSampleDataLoader.num_workers
NegSampleDataLoader.pin_memory
NegSampleDataLoader.pin_memory_device
NegSampleDataLoader.prefetch_factor
NegSampleDataLoader.sampler
NegSampleDataLoader.timeout
- recbole.data.dataloader.general_dataloader
FullSortEvalDataLoader
FullSortEvalDataLoader.batch_size
FullSortEvalDataLoader.collate_fn()
FullSortEvalDataLoader.dataset
FullSortEvalDataLoader.drop_last
FullSortEvalDataLoader.num_workers
FullSortEvalDataLoader.pin_memory
FullSortEvalDataLoader.pin_memory_device
FullSortEvalDataLoader.prefetch_factor
FullSortEvalDataLoader.sampler
FullSortEvalDataLoader.timeout
FullSortEvalDataLoader.update_config()
NegSampleEvalDataLoader
NegSampleEvalDataLoader.batch_size
NegSampleEvalDataLoader.collate_fn()
NegSampleEvalDataLoader.dataset
NegSampleEvalDataLoader.drop_last
NegSampleEvalDataLoader.num_workers
NegSampleEvalDataLoader.pin_memory
NegSampleEvalDataLoader.pin_memory_device
NegSampleEvalDataLoader.prefetch_factor
NegSampleEvalDataLoader.sampler
NegSampleEvalDataLoader.timeout
NegSampleEvalDataLoader.update_config()
TrainDataLoader
TrainDataLoader.batch_size
TrainDataLoader.collate_fn()
TrainDataLoader.dataset
TrainDataLoader.drop_last
TrainDataLoader.num_workers
TrainDataLoader.pin_memory
TrainDataLoader.pin_memory_device
TrainDataLoader.prefetch_factor
TrainDataLoader.sampler
TrainDataLoader.timeout
TrainDataLoader.update_config()
- recbole.data.dataloader.knowledge_dataloader
KGDataLoader
KnowledgeBasedDataLoader
- recbole.data.dataloader.user_dataloader
UserDataLoader
- recbole.data.dataset
- recbole.data.customized_dataset
DIENDataset
GRU4RecKGDataset
KSRDataset
- recbole.data.dataset
Dataset
Dataset.dataset_name
Dataset.dataset_path
Dataset.field2type
Dataset.field2source
Dataset.field2id_token
Dataset.field2token_id
Dataset.field2seqlen
Dataset.uid_field
Dataset.iid_field
Dataset.label_field
Dataset.time_field
Dataset.inter_feat
Dataset.user_feat
Dataset.item_feat
Dataset.feat_name_list
Dataset.avg_actions_of_items
Dataset.avg_actions_of_users
Dataset.build()
Dataset.copy()
Dataset.copy_field_property()
Dataset.counter()
Dataset.field2feats()
Dataset.fields()
Dataset.float_like_fields
Dataset.get_item_feature()
Dataset.get_preload_weight()
Dataset.get_user_feature()
Dataset.history_item_matrix()
Dataset.history_user_matrix()
Dataset.id2token()
Dataset.inter_matrix()
Dataset.inter_num
Dataset.item_counter
Dataset.item_num
Dataset.join()
Dataset.leave_one_out()
Dataset.non_seq_fields
Dataset.num()
Dataset.save()
Dataset.seq_fields
Dataset.set_field_property()
Dataset.shuffle()
Dataset.sort()
Dataset.sparsity
Dataset.split_by_ratio()
Dataset.time_based_split()
Dataset.token2id()
Dataset.token_like_fields
Dataset.user_counter
Dataset.user_num
- recbole.data.kg_dataset
KnowledgeBasedDataset
KnowledgeBasedDataset.head_entity_field
KnowledgeBasedDataset.tail_entity_field
KnowledgeBasedDataset.relation_field
KnowledgeBasedDataset.entity_field
KnowledgeBasedDataset.kg_feat
KnowledgeBasedDataset.item2entity
KnowledgeBasedDataset.entity2item
KnowledgeBasedDataset.ckg_graph()
KnowledgeBasedDataset.entities
KnowledgeBasedDataset.entity_num
KnowledgeBasedDataset.head_entities
KnowledgeBasedDataset.kg_graph()
KnowledgeBasedDataset.relation_num
KnowledgeBasedDataset.relations
KnowledgeBasedDataset.tail_entities
- recbole.data.kg_seq_dataset
KGSeqDataset
- recbole.data.sequential_dataset
SequentialDataset
- recbole.data.interaction
Interaction
cat_interactions()
- recbole.data.utils
create_dataset()
create_samplers()
data_preparation()
get_dataloader()
load_split_dataloaders()
save_split_dataloaders()