Index

_

_normalize() (in module crslab.model.utils.modules.transformer) _starts() (crslab.model.conversation.transformer.transformer.TransformerModel method) (crslab.model.crs.kbrd.kbrd.KBRDModel method) (crslab.model.crs.kgsf.kgsf.KGSFModel method)

A

add_start_end_token_idx() (in module crslab.data.dataloader.utils) attention_dropout (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute)

attention_probs_dropout_prob (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute)

B

BaseDataLoader (class in crslab.data.dataloader.base) BaseModel (class in crslab.model.base) batch_size (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute) BERTModel (class in crslab.model.recommendation.bert.bert) build() (in module crslab.download) build_model() (crslab.model.base.BaseModel method) (crslab.model.conversation.gpt2.gpt2.GPT2Model method) (crslab.model.conversation.transformer.transformer.TransformerModel method) (crslab.model.crs.kbrd.kbrd.KBRDModel method) (crslab.model.crs.kgsf.kgsf.KGSFModel method) (crslab.model.crs.redial.redial_conv.ReDialConvModel method) (crslab.model.crs.redial.redial_rec.ReDialRecModel method) (crslab.model.crs.tgredial.tg_conv.TGConvModel method) (crslab.model.crs.tgredial.tg_policy.TGPolicyModel method) (crslab.model.crs.tgredial.tg_rec.TGRecModel method) (crslab.model.policy.conv_bert.conv_bert.ConvBERTModel method) (crslab.model.policy.mgcg.mgcg.MGCGModel method) (crslab.model.policy.pmi.pmi.PMIModel method) (crslab.model.policy.profile_bert.profile_bert.ProfileBERTModel method) (crslab.model.policy.topic_bert.topic_bert.TopicBERTModel method) (crslab.model.recommendation.bert.bert.BERTModel method) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel method) (crslab.model.recommendation.popularity.popularity.PopularityModel method) (crslab.model.recommendation.sasrec.modules.SASRec method) (crslab.model.recommendation.sasrec.sasrec.SASRECModel method) (crslab.model.recommendation.textcnn.textcnn.TextCNNModel method)

C

calculate_loss() (crslab.model.conversation.gpt2.gpt2.GPT2Model method) (crslab.model.crs.tgredial.tg_conv.TGConvModel method) check_build() (in module crslab.download) checksum() (crslab.download.DownloadableFile method) compute_loss() (crslab.model.recommendation.sasrec.modules.SASRec method) Config (class in crslab.config.config) context_truncate (crslab.model.conversation.gpt2.gpt2.GPT2Model attribute) (crslab.model.crs.tgredial.tg_conv.TGConvModel attribute) conv_and_pool() (crslab.model.recommendation.textcnn.textcnn.TextCNNModel method) conv_batchify() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.kbrd.KBRDDataLoader method) (crslab.data.dataloader.kgsf.KGSFDataLoader method) (crslab.data.dataloader.redial.ReDialDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method) conv_interact() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method) conv_process_fn() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.kbrd.KBRDDataLoader method) (crslab.data.dataloader.kgsf.KGSFDataLoader method) (crslab.data.dataloader.redial.ReDialDataLoader method) ConvBERTModel (class in crslab.model.policy.conv_bert.conv_bert) converse() (crslab.model.base.BaseModel method) (crslab.model.crs.kbrd.kbrd.KBRDModel method) (crslab.model.crs.kgsf.kgsf.KGSFModel method) create_position_codes() (in module crslab.model.utils.modules.transformer) cross_entropy() (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel method) (crslab.model.recommendation.sasrec.modules.SASRec method) crslab module crslab.config module crslab.config.config module crslab.data module crslab.data.dataloader module crslab.data.dataloader.base module crslab.data.dataloader.kbrd module crslab.data.dataloader.kgsf module crslab.data.dataloader.redial module crslab.data.dataloader.tgredial module crslab.data.dataloader.utils module crslab.data.dataset module crslab.data.dataset.durecdial module crslab.data.dataset.durecdial.durecdial module crslab.data.dataset.durecdial.resources module crslab.data.dataset.gorecdial module crslab.data.dataset.gorecdial.gorecdial module crslab.data.dataset.gorecdial.resources module crslab.data.dataset.inspired module crslab.data.dataset.inspired.inspired module crslab.data.dataset.inspired.resources module crslab.data.dataset.opendialkg module crslab.data.dataset.opendialkg.opendialkg module crslab.data.dataset.opendialkg.resources module crslab.data.dataset.redial module crslab.data.dataset.redial.redial module crslab.data.dataset.redial.resources module crslab.data.dataset.tgredial module crslab.data.dataset.tgredial.resources module crslab.data.dataset.tgredial.tgredial module crslab.download module crslab.model module crslab.model.base module crslab.model.conversation module

crslab.model.conversation.gpt2 module crslab.model.conversation.gpt2.gpt2 module crslab.model.conversation.transformer module crslab.model.conversation.transformer.transformer module crslab.model.crs module crslab.model.crs.kbrd module crslab.model.crs.kbrd.kbrd module crslab.model.crs.kgsf module crslab.model.crs.kgsf.kgsf module crslab.model.crs.kgsf.modules module crslab.model.crs.kgsf.resources module crslab.model.crs.redial module crslab.model.crs.redial.modules module crslab.model.crs.redial.redial_conv module crslab.model.crs.redial.redial_rec module crslab.model.crs.tgredial module crslab.model.crs.tgredial.tg_conv module crslab.model.crs.tgredial.tg_policy module crslab.model.crs.tgredial.tg_rec module crslab.model.policy module crslab.model.policy.conv_bert module crslab.model.policy.conv_bert.conv_bert module crslab.model.policy.mgcg module crslab.model.policy.mgcg.mgcg module crslab.model.policy.pmi module crslab.model.policy.pmi.pmi module crslab.model.policy.profile_bert module crslab.model.policy.profile_bert.profile_bert module crslab.model.policy.topic_bert module crslab.model.policy.topic_bert.topic_bert module crslab.model.recommendation module crslab.model.recommendation.bert module crslab.model.recommendation.bert.bert module crslab.model.recommendation.gru4rec module crslab.model.recommendation.gru4rec.gru4rec module crslab.model.recommendation.popularity module crslab.model.recommendation.popularity.popularity module crslab.model.recommendation.sasrec module crslab.model.recommendation.sasrec.modules module crslab.model.recommendation.sasrec.sasrec module crslab.model.recommendation.textcnn module crslab.model.recommendation.textcnn.textcnn module crslab.model.utils module crslab.model.utils.functions module crslab.model.utils.modules module crslab.model.utils.modules.attention module crslab.model.utils.modules.transformer module

D

dataloader_register_table (in module crslab.data) dataset_language_map (in module crslab.data) DATASET_PATH (in module crslab.config) dataset_register_table (in module crslab.data) decode_beam_search() (crslab.model.crs.kbrd.kbrd.KBRDModel method) decode_forced() (crslab.model.crs.kbrd.kbrd.KBRDModel method) decode_greedy() (crslab.model.crs.kbrd.kbrd.KBRDModel method) decoder_embedding_dim (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) decoder_hidden_size (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) decoder_num_layers (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) dialog_encoder_hidden_size (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) dialog_encoder_num_layers (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute)

download() (in module crslab.download) download_file() (crslab.download.DownloadableFile method) download_from_google_drive() (in module crslab.download) DownloadableFile (class in crslab.download) dropout (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) (crslab.model.recommendation.textcnn.textcnn.TextCNNModel attribute) dropout_hidden (crslab.model.policy.mgcg.mgcg.MGCGModel attribute) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute) dropout_input (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute) DuRecDialDataset (class in crslab.data.dataset.durecdial.durecdial)

E

edge_to_pyg_format() (in module crslab.model.utils.functions) embed (crslab.model.recommendation.textcnn.textcnn.TextCNNModel attribute) embedding_dim (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) (crslab.model.policy.mgcg.mgcg.MGCGModel attribute) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute) EMBEDDING_PATH (in module crslab.config) Embeddings (class in crslab.model.recommendation.sasrec.modules) embeddings_scale (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute)

encode_user() (crslab.model.crs.kbrd.kbrd.KBRDModel method) Encoder (class in crslab.model.recommendation.sasrec.modules) end_token_idx (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute)

F

ffn_size (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) filter_sizes (crslab.model.recommendation.textcnn.textcnn.TextCNNModel attribute) forward() (crslab.model.conversation.gpt2.gpt2.GPT2Model method) (crslab.model.conversation.transformer.transformer.TransformerModel method) (crslab.model.crs.kbrd.kbrd.KBRDModel method) (crslab.model.crs.kgsf.kgsf.KGSFModel method) (crslab.model.crs.kgsf.modules.GateLayer method) (crslab.model.crs.kgsf.modules.TransformerDecoderKG method) (crslab.model.crs.kgsf.modules.TransformerDecoderLayerKG method) (crslab.model.crs.redial.modules.HRNN method) (crslab.model.crs.redial.modules.SwitchingDecoder method) (crslab.model.crs.redial.redial_conv.ReDialConvModel method) (crslab.model.crs.redial.redial_rec.ReDialRecModel method) (crslab.model.crs.tgredial.tg_conv.TGConvModel method) (crslab.model.crs.tgredial.tg_policy.TGPolicyModel method) (crslab.model.crs.tgredial.tg_rec.TGRecModel method) (crslab.model.policy.conv_bert.conv_bert.ConvBERTModel method) (crslab.model.policy.mgcg.mgcg.MGCGModel method) (crslab.model.policy.pmi.pmi.PMIModel method) (crslab.model.policy.profile_bert.profile_bert.ProfileBERTModel method) (crslab.model.policy.topic_bert.topic_bert.TopicBERTModel method) (crslab.model.recommendation.bert.bert.BERTModel method) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel method) (crslab.model.recommendation.popularity.popularity.PopularityModel method) (crslab.model.recommendation.sasrec.modules.Embeddings method) (crslab.model.recommendation.sasrec.modules.Encoder method) (crslab.model.recommendation.sasrec.modules.Intermediate method) (crslab.model.recommendation.sasrec.modules.Layer method) (crslab.model.recommendation.sasrec.modules.LayerNorm method) (crslab.model.recommendation.sasrec.modules.SASRec method) (crslab.model.recommendation.sasrec.modules.SelfAttention method) (crslab.model.recommendation.sasrec.sasrec.SASRECModel method) (crslab.model.recommendation.textcnn.textcnn.TextCNNModel method) (crslab.model.utils.modules.attention.SelfAttentionBatch method) (crslab.model.utils.modules.attention.SelfAttentionSeq method) (crslab.model.utils.modules.transformer.MultiHeadAttention method) (crslab.model.utils.modules.transformer.TransformerDecoder method) (crslab.model.utils.modules.transformer.TransformerDecoderLayer method) (crslab.model.utils.modules.transformer.TransformerEncoder method) (crslab.model.utils.modules.transformer.TransformerEncoderLayer method) (crslab.model.utils.modules.transformer.TransformerFFN method)

freeze_parameters() (crslab.model.crs.kbrd.kbrd.KBRDModel method) (crslab.model.crs.kgsf.kgsf.KGSFModel method)

G

GateLayer (class in crslab.model.crs.kgsf.modules) gelu() (in module crslab.model.recommendation.sasrec.modules) generate() (crslab.model.conversation.gpt2.gpt2.GPT2Model method) (crslab.model.crs.tgredial.tg_conv.TGConvModel method) generate_bs() (crslab.model.conversation.gpt2.gpt2.GPT2Model method) (crslab.model.crs.tgredial.tg_conv.TGConvModel method) get() (crslab.config.config.Config method) get_conv_data() (crslab.data.dataloader.base.BaseDataLoader method) get_data() (crslab.data.dataloader.base.BaseDataLoader method) get_dataloader() (in module crslab.data) get_dataset() (in module crslab.data)

get_length() (crslab.model.policy.mgcg.mgcg.MGCGModel method) get_model() (in module crslab.model) get_onehot() (in module crslab.data.dataloader.utils) get_policy_data() (crslab.data.dataloader.base.BaseDataLoader method) get_pretrain_data() (crslab.data.dataloader.kgsf.KGSFDataLoader method) get_rec_data() (crslab.data.dataloader.base.BaseDataLoader method) get_utterance_encoding() (crslab.model.crs.redial.modules.HRNN method) GoRecDialDataset (class in crslab.data.dataset.gorecdial.gorecdial) GPT2Model (class in crslab.model.conversation.gpt2.gpt2) GRU4RECModel (class in crslab.model.recommendation.gru4rec.gru4rec) guide() (crslab.model.base.BaseModel method)

H

hidden_act (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute) hidden_dropout_prob (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute)

hidden_size (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.policy.mgcg.mgcg.MGCGModel attribute) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute) HRNN (class in crslab.model.crs.redial.modules)

I

init_model() (crslab.model.recommendation.sasrec.modules.SASRec method) init_sas_weights() (crslab.model.recommendation.sasrec.modules.SASRec method) initializer_range (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute) InspiredDataset (class in crslab.data.dataset.inspired.inspired)

Intermediate (class in crslab.model.recommendation.sasrec.modules) item_size (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.bert.bert.BERTModel attribute) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute) (crslab.model.recommendation.popularity.popularity.PopularityModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute)

K

KBRDDataLoader (class in crslab.data.dataloader.kbrd) KBRDModel (class in crslab.model.crs.kbrd.kbrd) kg_emb_dim (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute)

KGSFDataLoader (class in crslab.data.dataloader.kgsf) KGSFModel (class in crslab.model.crs.kgsf.kgsf)

L

Layer (class in crslab.model.recommendation.sasrec.modules) layer_sizes (crslab.model.crs.redial.redial_rec.ReDialRecModel attribute) LayerNorm (class in crslab.model.recommendation.sasrec.modules) learn_positional_embeddings (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute)

load_model() (crslab.model.recommendation.sasrec.modules.SASRec method) load_yaml_configs() (crslab.config.config.Config static method) longest_label (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute)

M

make_dir() (in module crslab.download) mark_done() (in module crslab.download) max_seq_length (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute) merge_utt() (in module crslab.data.dataloader.utils) merge_utt_replace() (in module crslab.data.dataloader.utils) MGCGModel (class in crslab.model.policy.mgcg.mgcg) MODEL_PATH (in module crslab.config) module crslab crslab.config crslab.config.config crslab.data crslab.data.dataloader crslab.data.dataloader.base crslab.data.dataloader.kbrd crslab.data.dataloader.kgsf crslab.data.dataloader.redial crslab.data.dataloader.tgredial crslab.data.dataloader.utils crslab.data.dataset crslab.data.dataset.durecdial crslab.data.dataset.durecdial.durecdial crslab.data.dataset.durecdial.resources crslab.data.dataset.gorecdial crslab.data.dataset.gorecdial.gorecdial crslab.data.dataset.gorecdial.resources crslab.data.dataset.inspired crslab.data.dataset.inspired.inspired crslab.data.dataset.inspired.resources crslab.data.dataset.opendialkg crslab.data.dataset.opendialkg.opendialkg crslab.data.dataset.opendialkg.resources crslab.data.dataset.redial crslab.data.dataset.redial.redial crslab.data.dataset.redial.resources crslab.data.dataset.tgredial crslab.data.dataset.tgredial.resources crslab.data.dataset.tgredial.tgredial crslab.download crslab.model crslab.model.base crslab.model.conversation crslab.model.conversation.gpt2 crslab.model.conversation.gpt2.gpt2 crslab.model.conversation.transformer crslab.model.conversation.transformer.transformer crslab.model.crs crslab.model.crs.kbrd crslab.model.crs.kbrd.kbrd crslab.model.crs.kgsf crslab.model.crs.kgsf.kgsf crslab.model.crs.kgsf.modules crslab.model.crs.kgsf.resources crslab.model.crs.redial crslab.model.crs.redial.modules crslab.model.crs.redial.redial_conv crslab.model.crs.redial.redial_rec crslab.model.crs.tgredial crslab.model.crs.tgredial.tg_conv crslab.model.crs.tgredial.tg_policy crslab.model.crs.tgredial.tg_rec crslab.model.policy crslab.model.policy.conv_bert crslab.model.policy.conv_bert.conv_bert crslab.model.policy.mgcg crslab.model.policy.mgcg.mgcg crslab.model.policy.pmi crslab.model.policy.pmi.pmi crslab.model.policy.profile_bert crslab.model.policy.profile_bert.profile_bert crslab.model.policy.topic_bert crslab.model.policy.topic_bert.topic_bert crslab.model.recommendation crslab.model.recommendation.bert crslab.model.recommendation.bert.bert crslab.model.recommendation.gru4rec crslab.model.recommendation.gru4rec.gru4rec crslab.model.recommendation.popularity crslab.model.recommendation.popularity.popularity crslab.model.recommendation.sasrec crslab.model.recommendation.sasrec.modules crslab.model.recommendation.sasrec.sasrec crslab.model.recommendation.textcnn crslab.model.recommendation.textcnn.textcnn crslab.model.utils crslab.model.utils.functions crslab.model.utils.modules crslab.model.utils.modules.attention crslab.model.utils.modules.transformer

move() (in module crslab.download) movie_num (crslab.model.recommendation.textcnn.textcnn.TextCNNModel attribute) MultiHeadAttention (class in crslab.model.utils.modules.transformer)

N

n_entity (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_rec.ReDialRecModel attribute) n_heads (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) n_layers (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) n_positions (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) n_relation (crslab.model.crs.kbrd.kbrd.KBRDModel attribute)

n_sent (crslab.model.policy.mgcg.mgcg.MGCGModel attribute) (crslab.model.policy.profile_bert.profile_bert.ProfileBERTModel attribute) n_word (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) neginf() (in module crslab.model.utils.modules.transformer) num_attention_heads (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute) num_bases (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) num_filters (crslab.model.recommendation.textcnn.textcnn.TextCNNModel attribute) num_hidden_layers (crslab.model.crs.tgredial.tg_rec.TGRecModel attribute) (crslab.model.recommendation.sasrec.sasrec.SASRECModel attribute) num_layers (crslab.model.policy.mgcg.mgcg.MGCGModel attribute) (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel attribute)

O

OpenDialKGDataset (class in crslab.data.dataset.opendialkg.opendialkg)

P

pad_entity_idx (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_rec.ReDialRecModel attribute) pad_id (crslab.model.conversation.gpt2.gpt2.GPT2Model attribute) (crslab.model.crs.tgredial.tg_conv.TGConvModel attribute) pad_token_idx (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) pad_topic (crslab.model.policy.pmi.pmi.PMIModel attribute) pad_word_idx (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) padded_tensor() (in module crslab.data.dataloader.utils) PMIModel (class in crslab.model.policy.pmi.pmi) policy_batchify() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.kbrd.KBRDDataLoader method) (crslab.data.dataloader.kgsf.KGSFDataLoader method) (crslab.data.dataloader.redial.ReDialDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method)

policy_process_fn() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method) PopularityModel (class in crslab.model.recommendation.popularity.popularity) pretrain_batchify() (crslab.data.dataloader.kgsf.KGSFDataLoader method) pretrain_embedding (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) pretrain_infomax() (crslab.model.crs.kgsf.kgsf.KGSFModel method) PRETRAIN_PATH (in module crslab.config) pretrained_embedding (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) ProfileBERTModel (class in crslab.model.policy.profile_bert.profile_bert)

R

rec_batchify() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.kbrd.KBRDDataLoader method) (crslab.data.dataloader.kgsf.KGSFDataLoader method) (crslab.data.dataloader.redial.ReDialDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method) rec_interact() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method) rec_process_fn() (crslab.data.dataloader.base.BaseDataLoader method) (crslab.data.dataloader.kbrd.KBRDDataLoader method) (crslab.data.dataloader.kgsf.KGSFDataLoader method) (crslab.data.dataloader.redial.ReDialDataLoader method) (crslab.data.dataloader.tgredial.TGReDialDataLoader method) recommend() (crslab.model.base.BaseModel method) (crslab.model.crs.kbrd.kbrd.KBRDModel method) (crslab.model.crs.kgsf.kgsf.KGSFModel method)

reconstruct_input() (crslab.model.recommendation.gru4rec.gru4rec.GRU4RECModel method) ReDialConvModel (class in crslab.model.crs.redial.redial_conv) ReDialDataLoader (class in crslab.data.dataloader.redial) ReDialDataset (class in crslab.data.dataset.redial.redial) ReDialRecModel (class in crslab.model.crs.redial.redial_rec) reduction (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) relu_dropout (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) remove_dir() (in module crslab.download) response_truncate (crslab.model.conversation.gpt2.gpt2.GPT2Model attribute) (crslab.model.crs.tgredial.tg_conv.TGConvModel attribute) retain_recommender_target() (crslab.data.dataloader.base.BaseDataLoader method)

S

SASRec (class in crslab.model.recommendation.sasrec.modules) SASRECModel (class in crslab.model.recommendation.sasrec.sasrec) save_model() (crslab.model.recommendation.sasrec.modules.SASRec method) SAVE_PATH (in module crslab.config) SelfAttention (class in crslab.model.recommendation.sasrec.modules) SelfAttentionBatch (class in crslab.model.utils.modules.attention) SelfAttentionSeq (class in crslab.model.utils.modules.attention)

sort_for_packed_sequence() (in module crslab.model.utils.functions) start_token_idx (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) swish() (in module crslab.model.recommendation.sasrec.modules) SwitchingDecoder (class in crslab.model.crs.redial.modules)

T

test_data (crslab.data.dataset.durecdial.durecdial.DuRecDialDataset attribute) (crslab.data.dataset.gorecdial.gorecdial.GoRecDialDataset attribute) (crslab.data.dataset.inspired.inspired.InspiredDataset attribute) (crslab.data.dataset.opendialkg.opendialkg.OpenDialKGDataset attribute) (crslab.data.dataset.redial.redial.ReDialDataset attribute) (crslab.data.dataset.tgredial.tgredial.TGReDialDataset attribute) TextCNNModel (class in crslab.model.recommendation.textcnn.textcnn) TGConvModel (class in crslab.model.crs.tgredial.tg_conv) TGPolicyModel (class in crslab.model.crs.tgredial.tg_policy) TGRecModel (class in crslab.model.crs.tgredial.tg_rec) TGReDialDataLoader (class in crslab.data.dataloader.tgredial) TGReDialDataset (class in crslab.data.dataset.tgredial.tgredial) token_emb_dim (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) topic_class_num (crslab.model.policy.conv_bert.conv_bert.ConvBERTModel attribute) (crslab.model.policy.mgcg.mgcg.MGCGModel attribute) (crslab.model.policy.pmi.pmi.PMIModel attribute) (crslab.model.policy.profile_bert.profile_bert.ProfileBERTModel attribute) (crslab.model.policy.topic_bert.topic_bert.TopicBERTModel attribute)

TopicBERTModel (class in crslab.model.policy.topic_bert.topic_bert) train_data (crslab.data.dataset.durecdial.durecdial.DuRecDialDataset attribute) (crslab.data.dataset.gorecdial.gorecdial.GoRecDialDataset attribute) (crslab.data.dataset.inspired.inspired.InspiredDataset attribute) (crslab.data.dataset.opendialkg.opendialkg.OpenDialKGDataset attribute) (crslab.data.dataset.redial.redial.ReDialDataset attribute) (crslab.data.dataset.tgredial.tgredial.TGReDialDataset attribute) TransformerDecoder (class in crslab.model.utils.modules.transformer) TransformerDecoderKG (class in crslab.model.crs.kgsf.modules) TransformerDecoderLayer (class in crslab.model.utils.modules.transformer) TransformerDecoderLayerKG (class in crslab.model.crs.kgsf.modules) TransformerEncoder (class in crslab.model.utils.modules.transformer) TransformerEncoderLayer (class in crslab.model.utils.modules.transformer) TransformerFFN (class in crslab.model.utils.modules.transformer) TransformerModel (class in crslab.model.conversation.transformer.transformer) transpose_for_scores() (crslab.model.recommendation.sasrec.modules.SelfAttention method) truncate() (in module crslab.data.dataloader.utils)

U

unk_token_idx (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) untar() (in module crslab.download) use_dropout (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute)

user_emb_dim (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) user_proj_dim (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) utterance_encoder_hidden_size (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute)

V

valid_data (crslab.data.dataset.durecdial.durecdial.DuRecDialDataset attribute) (crslab.data.dataset.gorecdial.gorecdial.GoRecDialDataset attribute) (crslab.data.dataset.inspired.inspired.InspiredDataset attribute) (crslab.data.dataset.opendialkg.opendialkg.OpenDialKGDataset attribute) (crslab.data.dataset.redial.redial.ReDialDataset attribute) (crslab.data.dataset.tgredial.tgredial.TGReDialDataset attribute) vocab (crslab.data.dataset.durecdial.durecdial.DuRecDialDataset attribute) (crslab.data.dataset.gorecdial.gorecdial.GoRecDialDataset attribute) (crslab.data.dataset.inspired.inspired.InspiredDataset attribute) (crslab.data.dataset.opendialkg.opendialkg.OpenDialKGDataset attribute) (crslab.data.dataset.redial.redial.ReDialDataset attribute) (crslab.data.dataset.tgredial.tgredial.TGReDialDataset attribute)

vocab_size (crslab.model.conversation.transformer.transformer.TransformerModel attribute) (crslab.model.crs.kbrd.kbrd.KBRDModel attribute) (crslab.model.crs.kgsf.kgsf.KGSFModel attribute) (crslab.model.crs.redial.redial_conv.ReDialConvModel attribute) (crslab.model.policy.mgcg.mgcg.MGCGModel attribute)