Modules

• ReDIAL
  • Recommender Module
    • RedialRecConfig
      • RedialRecConfig.__init__()
    • RedialRecTokenizer
      • RedialRecTokenizer.__init__()
      • RedialRecTokenizer.get_init_kwargs()
      • RedialRecTokenizer.load_from_dataset()
      • RedialRecTokenizer.preprocess()
      • RedialRecTokenizer.encode_plus()
      • RedialRecTokenizer.batch_encode_plus()
      • RedialRecTokenizer.process_entities()
      • RedialRecTokenizer.encodes()
      • RedialRecTokenizer.decode()
    • RedialRec
      • RedialRec.__init__()
      • RedialRec.forward()
      • RedialRec.response()
  • Generator Module
    • RedialGenConfig
      • RedialGenConfig.__init__()
    • RedialGenTokenizer
      • RedialGenTokenizer.__init__()
      • RedialGenTokenizer.get_init_kwargs()
      • RedialGenTokenizer.load_from_dataset()
      • RedialGenTokenizer.preprocess()
      • RedialGenTokenizer.encode_plus()
      • RedialGenTokenizer.batch_encode_plus()
      • RedialGenTokenizer.process_entities()
      • RedialGenTokenizer.encodes()
      • RedialGenTokenizer.tokenize()
      • RedialGenTokenizer.vocab_size
      • RedialGenTokenizer.decode()
    • RedialGen
      • RedialGen.__init__()
      • RedialGen.forward()
      • RedialGen.response()
  • Supporting Modules
    • AutoRec
      • AutoRec.__init__()
      • AutoRec.load_checkpoint()
      • AutoRec.forward()
    • UserEncoder
      • UserEncoder.__init__()
      • UserEncoder.forward()
    • ReconstructionLoss
      • ReconstructionLoss.__init__()
      • ReconstructionLoss.forward()
      • ReconstructionLoss.normalize_loss_reset()
    • get_best_beam()
    • n_gram_repeats()
    • Beam
      • Beam.__init__()
      • Beam.__str__()
      • Beam.normalized_score()
    • BeamSearch
      • BeamSearch.update_beams()
    • HRNN
      • HRNN.__init__()
      • HRNN.forward()
    • HRNNForClassification
      • HRNNForClassification.__init__()
      • HRNNForClassification.forward()
    • RedialSentimentAnalysisLoss
      • RedialSentimentAnalysisLoss.__init__()
      • RedialSentimentAnalysisLoss.forward()
    • DecoderGRU
      • DecoderGRU.__init__()
      • DecoderGRU.set_pretrained_embeddings()
      • DecoderGRU.forward()
    • SwitchingDecoder
      • SwitchingDecoder.__init__()
      • SwitchingDecoder.set_pretrained_embeddings()
      • SwitchingDecoder.forward()
      • SwitchingDecoder.replace_movie_with_words()
      • SwitchingDecoder.generate()
    • NLTKTokenizer
      • NLTKTokenizer.__init__()
    • get_tokenizer()
    • RnnTokenizer()
• KBRD
  • Recommender Module
    • KBRDRecConfig
    • KBRDRecTokenizer
    • KBRDRec
  • Generator Module
    • KBRDGenConfig
    • KBRDGenTokenizer
    • KBRDGen
  • Supporting Modules
    • NLTKTokenizer
    • get_tokenizer()
    • KBRDWordTokenizer()
    • TorchGeneratorModel
      • TorchGeneratorModel.__init__()
      • TorchGeneratorModel._starts()
      • TorchGeneratorModel.decode_greedy()
      • TorchGeneratorModel.decode_forced()
      • TorchGeneratorModel.reorder_encoder_states()
      • TorchGeneratorModel.reorder_decoder_incremental_state()
      • TorchGeneratorModel.forward()
    • TransformerEncoder
      • TransformerEncoder.__init__()
      • TransformerEncoder.forward()
    • TransformerEncoderLayer
      • TransformerEncoderLayer.__init__()
      • TransformerEncoderLayer.forward()
    • TransformerDecoder
      • TransformerDecoder.__init__()
      • TransformerDecoder.forward()
    • TransformerDecoderLayer
      • TransformerDecoderLayer.__init__()
      • TransformerDecoderLayer.forward()
    • TransformerGeneratorModel
      • TransformerGeneratorModel.__init__()
      • TransformerGeneratorModel.reorder_encoder_states()
    • BasicAttention
      • BasicAttention.__init__()
      • BasicAttention.forward()
    • MultiHeadAttention
      • MultiHeadAttention.__init__()
      • MultiHeadAttention.forward()
    • TransformerFFN
      • TransformerFFN.__init__()
      • TransformerFFN.forward()
• KGSF
  • Recommender Module
    • KGSFRecConfig
    • KGSFRecTokenizer
      • KGSFRecTokenizer.__init__()
      • KGSFRecTokenizer.get_init_kwargs()
      • KGSFRecTokenizer.padding_w2v()
      • KGSFRecTokenizer.padding_context()
      • KGSFRecTokenizer.encode()
      • KGSFRecTokenizer.decode()
    • KGSFRec
      • KGSFRec.__init__()
      • KGSFRec.forward()
      • KGSFRec.response()
  • Generator Module
    • KGSFGenConfig
    • KGSFGenTokenizer
      • KGSFGenTokenizer.__init__()
      • KGSFGenTokenizer.get_init_kwargs()
      • KGSFGenTokenizer.padding_w2v()
      • KGSFGenTokenizer.padding_context()
      • KGSFGenTokenizer.encode()
      • KGSFGenTokenizer.decode()
    • KGSFGen
      • KGSFGen.__init__()
      • KGSFGen._starts()
      • KGSFGen.decode_greedy()
      • KGSFGen.decode_forced()
      • KGSFGen.forward()
      • KGSFGen.response()
  • Supporting Modules
    • kaiming_reset_parameters()
    • GraphConvolution
      • GraphConvolution.__init__()
      • GraphConvolution.forward()
      • GraphConvolution.__repr__()
    • GCN
      • GCN.__init__()
      • GCN.forward()
    • GraphAttentionLayer
      • GraphAttentionLayer.__init__()
      • GraphAttentionLayer.forward()
      • GraphAttentionLayer.__repr__()
    • SelfAttentionLayer
      • SelfAttentionLayer.__init__()
      • SelfAttentionLayer.forward()
    • SelfAttentionLayer_batch
      • SelfAttentionLayer_batch.__init__()
      • SelfAttentionLayer_batch.forward()
    • SelfAttentionLayer2
      • SelfAttentionLayer2.__init__()
      • SelfAttentionLayer2.forward()
    • BiAttention
      • BiAttention.__init__()
      • BiAttention.forward()
    • GAT
      • GAT.__init__()
      • GAT.forward()
    • SpecialSpmmFunction
      • SpecialSpmmFunction.forward()
      • SpecialSpmmFunction.backward()
    • SpecialSpmm
      • SpecialSpmm.forward()
    • SpGraphAttentionLayer
      • SpGraphAttentionLayer.__init__()
      • SpGraphAttentionLayer.forward()
      • SpGraphAttentionLayer.__repr__()
    • SpGAT
      • SpGAT.__init__()
      • SpGAT.forward()
    • _add_neighbors()
    • _normalize()
    • _build_encoder()
    • _build_encoder4kg()
    • _build_encoder_mask()
    • _build_decoder()
    • _build_decoder4kg()
    • create_position_codes()
    • BasicAttention
      • BasicAttention.__init__()
      • BasicAttention.forward()
    • MultiHeadAttention
      • MultiHeadAttention.__init__()
      • MultiHeadAttention.forward()
    • TransformerFFN
      • TransformerFFN.__init__()
      • TransformerFFN.forward()
    • TransformerResponseWrapper
      • TransformerResponseWrapper.__init__()
      • TransformerResponseWrapper.forward()
    • TransformerEncoder4kg
      • TransformerEncoder4kg.__init__()
      • TransformerEncoder4kg.forward()
    • TransformerEncoderLayer
      • TransformerEncoderLayer.__init__()
      • TransformerEncoderLayer.forward()
    • TransformerEncoder
      • TransformerEncoder.__init__()
      • TransformerEncoder.forward()
    • TransformerEncoder_mask
      • TransformerEncoder_mask.__init__()
      • TransformerEncoder_mask.forward()
    • TransformerDecoderLayer
      • TransformerDecoderLayer.__init__()
      • TransformerDecoderLayer.forward()
    • TransformerDecoder
      • TransformerDecoder.__init__()
      • TransformerDecoder.forward()
    • TransformerDecoderLayerKG
      • TransformerDecoderLayerKG.__init__()
      • TransformerDecoderLayerKG.forward()
    • TransformerDecoder
      • TransformerDecoder.__init__()
      • TransformerDecoder.forward()
    • TransformerMemNetModel
      • TransformerMemNetModel.__init__()
      • TransformerMemNetModel.forward()
    • TorchGeneratorModel
      • TorchGeneratorModel.__init__()
      • TorchGeneratorModel._starts()
      • TorchGeneratorModel.decode_greedy()
      • TorchGeneratorModel.decode_forced()
      • TorchGeneratorModel.reorder_encoder_states()
      • TorchGeneratorModel.reorder_decoder_incremental_state()
      • TorchGeneratorModel.forward()
    • neginf()
    • _create_embeddings()
    • _create_entity_embeddings()
    • _edge_list()
    • _concept_edge_list4GCN()
    • seed_everything()
• UniCRS
  • Recommender Module
    • UnicrsRecConfig
      • UnicrsRecConfig.__init__()
    • UnicrsRecTokenizer
      • UnicrsRecTokenizer.__init__()
      • UnicrsRecTokenizer.__call__()
      • UnicrsRecTokenizer.load_from_dataset()
      • UnicrsRecTokenizer.mergeEncoding()
      • UnicrsRecTokenizer.encodes()
      • UnicrsRecTokenizer.decode()
  • Generator Module
    • UnicrsGenConfig
      • UnicrsGenConfig.__init__()
    • UnicrsGenTokenizer
      • UnicrsGenTokenizer.__init__()
      • UnicrsGenTokenizer.load_from_dataset()
      • UnicrsGenTokenizer.mergeEncoding()
      • UnicrsGenTokenizer.__call__()
      • UnicrsGenTokenizer.encodes()
    • UnicrsGen
      • UnicrsGen.__init__()
      • UnicrsGen.forward()
      • UnicrsGen.generate()
      • UnicrsGen.response()
  • Supporting Modules
    • KGPrompt
      • KGPrompt.__init__()
      • KGPrompt.forward()
    • GPT2Attention
      • GPT2Attention.__init__()
      • GPT2Attention._split_heads()
      • GPT2Attention._merge_heads()
      • GPT2Attention.forward()
    • GPT2Block
      • GPT2Block.__init__()
      • GPT2Block.forward()
    • GPT2Model
      • GPT2Model.__init__()
      • GPT2Model.get_input_embeddings()
      • GPT2Model.set_input_embeddings()
      • GPT2Model._prune_heads()
      • GPT2Model.forward()
    • PromptGPT2LMHead
      • PromptGPT2LMHead.__init__()
      • PromptGPT2LMHead.get_output_embeddings()
      • PromptGPT2LMHead.forward()
• LLMs
  • General LLM
    • LLMConfig
      • LLMConfig.answer_name
      • LLMConfig.answer_mask
      • LLMConfig.prompt
      • LLMConfig.model_name
      • LLMConfig.__init__()
    • LLMRecConfig
      • LLMRecConfig.answer_name
      • LLMRecConfig.answer_type
      • LLMRecConfig.prompt
      • LLMRecConfig.model_name
      • LLMRecConfig.__init__()
  • ChatGPT
    • ChatgptGen
      • ChatgptGen.__init__()
      • ChatgptGen.from_pretrained()
      • ChatgptGen.save_pretrained()
      • ChatgptGen.get_tokenizer()
      • ChatgptGen.response()
    • ChatgptRec
      • ChatgptRec.__init__()
      • ChatgptRec.from_pretrained()
      • ChatgptRec.save_pretrained()
      • ChatgptRec.get_tokenizer()
      • ChatgptRec.response()
    • ChatgptTokenizer
      • ChatgptTokenizer.__init__()
      • ChatgptTokenizer.__call__()
  • Llama
    • LlamaGen
      • LlamaGen.__init__()
      • LlamaGen.from_pretrained()
      • LlamaGen.save_pretrained()
      • LlamaGen.get_tokenizer()
      • LlamaGen.response()
    • LlamaTokenizer
      • LlamaTokenizer.__init__()
      • LlamaTokenizer.preprocess()