MCPcopy Create free account

hub / github.com/RUCDM/KB4Rec / functions

Functions275 in github.com/RUCDM/KB4Rec

⨍ Functions 275◇ Types & classes 18
↓ 84 callersMethodto
指定运行模式 :param device: cude or cpu :return:
Projects/UPGAN/code/util/Regularization.py:26
↓ 10 callersMethodquery_judge
(self, query_now, tails)
Projects/UPGAN/code/Model/UGAT.py:151
↓ 9 callersMethodnormLoss
(embeddings, dim=1)
Projects/UPGAN/code/Model/base_model.py:42
↓ 8 callersMethodform_query
(self, heads, rels, e1_embedded_user, pretrain=False)
Projects/UPGAN/code/Model/UGAT.py:122
↓ 7 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/UGAT.py:68
↓ 7 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/UGAT_mlp.py:69
↓ 6 callersMethodgen_sample
(self, heads, rels, neg_tails, n_sample=1, temperature=1.0, lambda_smooth=0.5)
Projects/UPGAN/code/train/trainer.py:213
↓ 6 callersMethodget_candidates
(self, pretrain=False)
Projects/UPGAN/code/Model/UGAT.py:93
↓ 5 callersMethodget_ent_all
(self)
Projects/UPGAN/code/Model/UGAT.py:96
↓ 5 callersMethodget_ent_all
(self)
Projects/UPGAN/code/Model/UGAT_mlp.py:96
↓ 5 callersMethodtrain
(self, start_epoch, end_epoch)
Projects/UPGAN/code/train/trainer.py:110
↓ 4 callersFunctionRGCN_kernal_new
(train_triple_list)
Projects/UPGAN/code/util/sage_kernel.py:45
↓ 4 callersMethodcreate_neg
(self, head, rel, num_sample=200)
Projects/UPGAN/code/util/dataset.py:114
↓ 4 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/dot_2layer.py:45
↓ 4 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/concat_1layer.py:44
↓ 4 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/DistMult.py:36
↓ 4 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/generator_concat.py:44
↓ 4 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/concat_2layer.py:45
↓ 4 callersMethodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/generator.py:44
↓ 4 callersFunctioneva_rank_list
(rank_list)
Projects/UPGAN/code/train/evaluation.py:8
↓ 4 callersMethodevaluate
(self, eval_loader, name)
Projects/UPGAN/code/train/evaluation.py:121
↓ 4 callersMethodfind_common
(self, gen_tails, right_tails, recall_list, recall_dict)
Projects/UPGAN/code/train/trainer.py:79
↓ 4 callersMethodloadEmbedding
(self, filename, model, cpu=False, alternative="model_state_dict")
Projects/UPGAN/code/pretrain/base_trainer.py:307
↓ 4 callersMethodloadEmbedding
(self, filename, model, cpu=False, alternative="model_state_dict")
Projects/UPGAN/code/train/base_trainer.py:309
↓ 4 callersMethodnorm_emb
(self)
Projects/UPGAN/code/Model/base_model.py:90
↓ 4 callersMethodnorm_one
(self)
Projects/UPGAN/code/Model/base_model.py:104
↓ 4 callersMethodnorm_user
(self)
Projects/UPGAN/code/Model/base_model.py:97
↓ 4 callersFunctionread_map
(filename)
Projects/UPGAN/code/util/utils.py:140
↓ 3 callersFunctioncreate_eval
(triple_list, all_hrt, relation_num, batch_size = 128)
Projects/UPGAN/code/train/load_data.py:14
↓ 3 callersMethodevaluate
(self, heads, rels, batch_rel_cand, pretrain=False)
Projects/UPGAN/code/Model/UGAT.py:99
↓ 3 callersMethodfetch_user_for_ent_new
(self, rgcn_kernel=None, rs_graph=None, kg_graph=None, pretrain=False)
Projects/UPGAN/code/Model/base_model.py:139
↓ 3 callersMethodget_emb_all
(self)
Projects/UPGAN/code/Model/dot_2layer.py:125
↓ 3 callersMethodget_emb_all
(self)
Projects/UPGAN/code/Model/concat_1layer.py:124
↓ 3 callersMethodget_emb_all
(self)
Projects/UPGAN/code/Model/generator_concat.py:121
↓ 3 callersMethodget_emb_all
(self)
Projects/UPGAN/code/Model/concat_2layer.py:125
↓ 3 callersMethodget_emb_all
(self)
Projects/UPGAN/code/Model/generator.py:124
↓ 3 callersMethodload_ckpt
(self, filename)
Projects/UPGAN/code/pretrain/base_trainer.py:272
↓ 3 callersMethodload_ckpt
(self, filename)
Projects/UPGAN/code/train/base_trainer.py:272
↓ 3 callersFunctionload_triple
(filename)
Projects/UPGAN/code/util/utils.py:183
↓ 3 callersMethodquery_add_noise
(self, query)
Projects/UPGAN/code/Model/dot_2layer.py:68
↓ 3 callersMethodquery_add_noise
(self, query)
Projects/UPGAN/code/Model/concat_1layer.py:67
↓ 3 callersMethodquery_add_noise
(self, query)
Projects/UPGAN/code/Model/generator_concat.py:66
↓ 3 callersMethodquery_add_noise
(self, query)
Projects/UPGAN/code/Model/concat_2layer.py:68
↓ 3 callersMethodquery_add_noise
(self, query)
Projects/UPGAN/code/Model/generator.py:67
↓ 3 callersMethodshow_norm
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:222
↓ 3 callersMethodshow_norm
(self)
Projects/UPGAN/code/train/base_trainer.py:222
↓ 2 callersMethodcalc_score
(self, head_rep, tail_rep, edge_list)
Projects/UPGAN/code/Model/layers.py:84
↓ 2 callersFunctioncheck
(mid, mids)
expansion/scan_triple.py:7
↓ 2 callersFunctioncheck
(mid, mids)
expansion/2step/scan_triple_2step.py:8
↓ 2 callersFunctioncreate_logger
(args)
Projects/UPGAN/code/util/utils.py:8
↓ 2 callersMethodcreate_neg_pop
(self, head, rel, num_sample=200)
Projects/UPGAN/code/util/dataset.py:97
↓ 2 callersFunctioneva_rank_list
(rank_list)
Projects/UPGAN/code/util/utils.py:123
↓ 2 callersMethodevaluate_single
(self, filename)
Projects/UPGAN/code/train/base_trainer.py:242
↓ 2 callersMethodfetch_user_batch
(self, rgcn_kernel, rs_graph, kg_graph, tp_hop, unreach_ids, batch_order, pretrain=False)
Projects/UPGAN/code/Model/base_model.py:110
↓ 2 callersMethodforward
(self, head_rep, tail_rep, tail_val, edge_list, rel_list, requires_grad=False)
Projects/UPGAN/code/Model/layers.py:121
↓ 2 callersMethodforward_kg
(self, e1, rel, all_e_ids=None)
Projects/UPGAN/code/Model/DistMult.py:112
↓ 2 callersMethodforward_rs
(self, head_rep, tail_rep, edge_list, rs_rel, requires_grad=False)
Projects/UPGAN/code/Model/layers.py:178
↓ 2 callersMethodforward_triple
:param heads: (B, ) :param rels: (B, ) :param tails: (B, N) :return:
Projects/UPGAN/code/Model/DistMult.py:64
↓ 2 callersMethodget_emb_all
(self)
Projects/UPGAN/code/Model/DistMult.py:109
↓ 2 callersMethodget_prior
(self, nege_tails)
Projects/UPGAN/code/train/trainer.py:208
↓ 2 callersMethodget_rgcn_kenel
(self, path_reserve, user_set)
Projects/UPGAN/code/pretrain/base_trainer.py:458
↓ 2 callersMethodget_rgcn_kenel
(self, path_reserve, user_set)
Projects/UPGAN/code/train/base_trainer.py:460
↓ 2 callersMethodget_user_all
(self, rgcn_kernel, pretrain=False)
Projects/UPGAN/code/Model/base_model.py:57
↓ 2 callersMethodget_weight
获得模型的权重列表 :param model: :return:
Projects/UPGAN/code/util/Regularization.py:41
↓ 2 callersFunctionload_dataset
(args, logger=None)
Projects/UPGAN/code/train/load_data.py:18
↓ 2 callersFunctionload_entity_dict
(train_triple_list, relation_num)
Projects/UPGAN/code/util/utils.py:31
↓ 2 callersFunctionload_head_dict
(train_triple_list, ent2hop, relation_num)
Projects/UPGAN/code/util/utils.py:67
↓ 2 callersFunctionload_rating
(filename, u_map, i_map)
Projects/UPGAN/code/util/utils.py:164
↓ 2 callersFunctionload_tail_dict
(train_triple_list, ent2hop, relation_num)
Projects/UPGAN/code/util/utils.py:83
↓ 2 callersFunctionmake_kernel_batch
(path_list, max_hop, ent_hop, hop_map)
Projects/UPGAN/code/util/triple_kernel.py:4
↓ 2 callersFunctionmake_kg_graph
(head_dict, max_hop, ent_hop, hop_map, sample_flag=False, num_sample=5)
Projects/UPGAN/code/util/kernel.py:50
↓ 2 callersFunctionmake_rs_graph_split
(ent_dict, node_tot, interaction, num_split=1, sample_flag=False, num_sample=5, logger=None)
Projects/UPGAN/code/util/kernel.py:4
↓ 2 callersFunctionprint_rank_list
(raw_list, fil_list, temp_str, logger)
Projects/UPGAN/code/train/evaluation.py:15
↓ 2 callersFunctionremap_entity
(max_hop, linked_num, entity_total, ent_hop)
Projects/UPGAN/code/util/utils.py:42
↓ 2 callersMethodsave_ckpt
(self)
Projects/UPGAN/code/pretrain/trainer.py:51
↓ 2 callersMethodset_edges
(self, edge_list_kg, edge_list_rs, remapped_entity_list, unreach_ents, rgcn_kenel)
Projects/UPGAN/code/Model/base_model.py:12
↓ 2 callersMethodshow_G_norm
(self)
Projects/UPGAN/code/train/trainer.py:102
↓ 2 callersFunctiontriple_list2hrt
(triple_list, relation_num)
Projects/UPGAN/code/util/utils.py:194
↓ 1 callersFunctionMakeEvalIterator
(r_ht, r_ht_all, batch_size)
Projects/UPGAN/code/util/utils.py:206
↓ 1 callersMethod__init__
(self, in_features, out_features, match_matrix, devidce, args)
Projects/UPGAN/code/Model/layers.py:70
↓ 1 callersMethodbatch_reserve_ents
(self, heads, path_list)
Projects/UPGAN/code/pretrain/base_trainer.py:434
↓ 1 callersMethodbatch_reserve_ents
(self, heads, path_list)
Projects/UPGAN/code/train/base_trainer.py:436
↓ 1 callersMethodcalc_entropy
(self, probs)
Projects/UPGAN/code/train/trainer.py:170
↓ 1 callersMethodcalc_scores
(self, heads, rels, sample_tails, all_tails, user_rep)
Projects/UPGAN/code/train/trainer.py:237
↓ 1 callersFunctioncount_pop
(triple_list, entity_num)
Projects/UPGAN/code/util/dataset.py:61
↓ 1 callersFunctioncreate_hr_t
(triple_list, relation_num)
Projects/UPGAN/code/util/dataset.py:50
↓ 1 callersFunctioncreate_train
(triple_list, entity_num, relation_num, args)
Projects/UPGAN/code/train/load_data.py:9
↓ 1 callersMethoddata_load_kg_rs
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:19
↓ 1 callersMethoddata_load_kg_rs
(self)
Projects/UPGAN/code/train/base_trainer.py:19
↓ 1 callersMethoddis_step
(self, heads, rels, tails, neg_tails, sample_weight, user_rep)
Projects/UPGAN/code/pretrain/trainer.py:100
↓ 1 callersMethoddis_step
(self, heads, rels, tails, neg_tails, sample_weight, user_rep)
Projects/UPGAN/code/train/trainer.py:245
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/dot_2layer.py:34
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/UGAT.py:58
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/concat_1layer.py:34
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/DistMult.py:30
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/generator_concat.py:34
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/concat_2layer.py:34
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/generator.py:34
↓ 1 callersMethodemb_def
(self)
Projects/UPGAN/code/Model/UGAT_mlp.py:59
↓ 1 callersMethodeval_sample_diversity
(self, epoch=0)
Projects/UPGAN/code/train/base_trainer.py:187
next →1–100 of 275, ranked by callers