recbole.dataΒΆ
- recbole.data.dataloader
- recbole.data.dataloader.abstract_dataloader
AbstractDataLoaderAbstractDataLoader._datasetAbstractDataLoader.shuffleAbstractDataLoader.prAbstractDataLoader.stepAbstractDataLoader._batch_sizeAbstractDataLoader.batch_sizeAbstractDataLoader.collate_fn()AbstractDataLoader.datasetAbstractDataLoader.drop_lastAbstractDataLoader.num_workersAbstractDataLoader.pin_memoryAbstractDataLoader.pin_memory_deviceAbstractDataLoader.prefetch_factorAbstractDataLoader.samplerAbstractDataLoader.set_batch_size()AbstractDataLoader.timeoutAbstractDataLoader.update_config()
NegSampleDataLoaderNegSampleDataLoader.batch_sizeNegSampleDataLoader.datasetNegSampleDataLoader.drop_lastNegSampleDataLoader.get_model()NegSampleDataLoader.num_workersNegSampleDataLoader.pin_memoryNegSampleDataLoader.pin_memory_deviceNegSampleDataLoader.prefetch_factorNegSampleDataLoader.samplerNegSampleDataLoader.timeout
- recbole.data.dataloader.general_dataloader
FullSortEvalDataLoaderFullSortEvalDataLoader.batch_sizeFullSortEvalDataLoader.collate_fn()FullSortEvalDataLoader.datasetFullSortEvalDataLoader.drop_lastFullSortEvalDataLoader.num_workersFullSortEvalDataLoader.pin_memoryFullSortEvalDataLoader.pin_memory_deviceFullSortEvalDataLoader.prefetch_factorFullSortEvalDataLoader.samplerFullSortEvalDataLoader.timeoutFullSortEvalDataLoader.update_config()
NegSampleEvalDataLoaderNegSampleEvalDataLoader.batch_sizeNegSampleEvalDataLoader.collate_fn()NegSampleEvalDataLoader.datasetNegSampleEvalDataLoader.drop_lastNegSampleEvalDataLoader.num_workersNegSampleEvalDataLoader.pin_memoryNegSampleEvalDataLoader.pin_memory_deviceNegSampleEvalDataLoader.prefetch_factorNegSampleEvalDataLoader.samplerNegSampleEvalDataLoader.timeoutNegSampleEvalDataLoader.update_config()
TrainDataLoaderTrainDataLoader.batch_sizeTrainDataLoader.collate_fn()TrainDataLoader.datasetTrainDataLoader.drop_lastTrainDataLoader.num_workersTrainDataLoader.pin_memoryTrainDataLoader.pin_memory_deviceTrainDataLoader.prefetch_factorTrainDataLoader.samplerTrainDataLoader.timeoutTrainDataLoader.update_config()
- recbole.data.dataloader.knowledge_dataloader
KGDataLoaderKnowledgeBasedDataLoader- recbole.data.dataloader.user_dataloader
UserDataLoader
- recbole.data.dataset
- recbole.data.customized_dataset
DIENDatasetGRU4RecKGDatasetKSRDataset- recbole.data.dataset
DatasetDataset.dataset_nameDataset.dataset_pathDataset.field2typeDataset.field2sourceDataset.field2id_tokenDataset.field2token_idDataset.field2seqlenDataset.uid_fieldDataset.iid_fieldDataset.label_fieldDataset.time_fieldDataset.inter_featDataset.user_featDataset.item_featDataset.feat_name_listDataset.avg_actions_of_itemsDataset.avg_actions_of_usersDataset.build()Dataset.copy()Dataset.copy_field_property()Dataset.counter()Dataset.field2feats()Dataset.fields()Dataset.float_like_fieldsDataset.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_numDataset.item_counterDataset.item_numDataset.join()Dataset.leave_one_out()Dataset.non_seq_fieldsDataset.num()Dataset.save()Dataset.seq_fieldsDataset.set_field_property()Dataset.shuffle()Dataset.sort()Dataset.sparsityDataset.split_by_ratio()Dataset.time_based_split()Dataset.token2id()Dataset.token_like_fieldsDataset.user_counterDataset.user_num
- recbole.data.kg_dataset
KnowledgeBasedDatasetKnowledgeBasedDataset.head_entity_fieldKnowledgeBasedDataset.tail_entity_fieldKnowledgeBasedDataset.relation_fieldKnowledgeBasedDataset.entity_fieldKnowledgeBasedDataset.kg_featKnowledgeBasedDataset.item2entityKnowledgeBasedDataset.entity2itemKnowledgeBasedDataset.ckg_graph()KnowledgeBasedDataset.entitiesKnowledgeBasedDataset.entity_numKnowledgeBasedDataset.head_entitiesKnowledgeBasedDataset.kg_graph()KnowledgeBasedDataset.relation_numKnowledgeBasedDataset.relationsKnowledgeBasedDataset.tail_entities
- recbole.data.kg_seq_dataset
KGSeqDataset- recbole.data.sequential_dataset
SequentialDataset
- recbole.data.interaction
Interactioncat_interactions()- recbole.data.utils
create_dataset()create_samplers()data_preparation()get_dataloader()load_split_dataloaders()save_split_dataloaders()