Generator Module

class recwizard.modules.kbrd.configuration_kbrd_gen.KBRDGenConfig(gen_dim: int | None = None, rec_dim: int | None = None, vocab_size: int | None = None, pad_idx: int | None = None, start_idx: int | None = None, end_idx: int | None = None, n_positions: int | None = None, n_heads: int | None = None, n_layers: int | None = None, ffn_size: int | None = None, dropout: float = 0, attention_dropout: float = 0, relu_dropout: float = 0, learn_positional_embeddings: bool = False, embeddings_scale: bool = True, rec_module_config: dict | None = None, **kwargs)

Configuration class to store the configuration of a KBRDGen.

class recwizard.modules.kbrd.tokenizer_kbrd_gen.KBRDGenTokenizer(vocab: List[str], id2entity: Dict[int, str] | None = None, **kwargs)

class recwizard.modules.kbrd.modeling_kbrd_gen.KBRDGen(config: KBRDGenConfig, **kwargs)

KBRDGen is a module that combines KBRDRec and TransformerGeneratorModel.