recbole.data.customized_dataset

We only recommend building customized datasets by inheriting.

Customized datasets named [Model Name]Dataset can be automatically called.

class recbole.data.dataset.customized_dataset.GRU4RecKGDataset(config, saved_dataset=None)[source]

Bases: recbole.data.dataset.kg_seq_dataset.Kg_Seq_Dataset

class recbole.data.dataset.customized_dataset.KSRDataset(config, saved_dataset=None)[source]

Bases: recbole.data.dataset.kg_seq_dataset.Kg_Seq_Dataset