Recommender Module#

class recwizard.modules.kbrd.configuration_kbrd_rec.KBRDRecConfig(n_entity: int | None = None, n_relation: int | None = None, sub_n_relation: int | None = None, dim: int | None = None, num_bases: int | None = None, **kwargs)[source]#: Configuration class to store the configuration of a KBRDRec.

class recwizard.modules.kbrd.tokenizer_kbrd_rec.KBRDRecTokenizer(entity2id: Dict[str, int] | None = None, id2entity: Dict[int, str] | None = None, pad_entity_id: int | None = None, tokenizers: List[PreTrainedTokenizerBase] | PreTrainedTokenizerBase | None = None, **kwargs)[source]#

class recwizard.modules.kbrd.modeling_kbrd_rec.KBRDRec(config: KBRDRecConfig, edge_index=None, edge_type=None, item_index=None, **kwargs)[source]#: KBRDRec is a module that implements the KBRD recommender.