MCPcopy Create free account

hub / github.com/RUCDM/KB4Rec / functions

Functions275 in github.com/RUCDM/KB4Rec

↓ 1 callersMethodeval_sample_quality
(self)
Projects/UPGAN/code/train/base_trainer.py:154
↓ 1 callersMethodevaluate_best
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:256
↓ 1 callersMethodevaluate_best
(self)
Projects/UPGAN/code/train/base_trainer.py:256
↓ 1 callersFunctionfilter_ent
(output)
expansion/2step/filter_ent.py:3
↓ 1 callersMethodforward
(self, e1, rel)
Projects/UPGAN/code/Model/dot_2layer.py:73
↓ 1 callersMethodforward
(self, e1, rel)
Projects/UPGAN/code/Model/concat_1layer.py:72
↓ 1 callersMethodforward
(self, e1, rel)
Projects/UPGAN/code/Model/DistMult.py:39
↓ 1 callersMethodforward
(self, e1, rel)
Projects/UPGAN/code/Model/generator_concat.py:71
↓ 1 callersMethodforward
(self, e1, rel)
Projects/UPGAN/code/Model/concat_2layer.py:73
↓ 1 callersMethodforward
(self, e1, rel)
Projects/UPGAN/code/Model/generator.py:72
↓ 1 callersMethodgen_step
(self, heads, rels, neg_tails, n_sample=1, temperature=1.0, lambda_smooth=0.5)
Projects/UPGAN/code/train/trainer.py:174
↓ 1 callersFunctiongetEn
(filename)
expansion/scan_triple.py:32
↓ 1 callersFunctiongetEn
(filename)
expansion/2step/scan_triple_2step.py:36
↓ 1 callersFunctiongetSub
(mids, filename2, filename3)
expansion/scan_triple.py:13
↓ 1 callersFunctiongetSub
(mids, filename2, filename3)
expansion/2step/scan_triple_2step.py:15
↓ 1 callersMethodget_ent_all
(self)
Projects/UPGAN/code/Model/base_model.py:87
↓ 1 callersMethodget_related_ents
(self, ent_now, path_list)
Projects/UPGAN/code/pretrain/base_trainer.py:411
↓ 1 callersMethodget_related_ents
(self, ent_now, path_list)
Projects/UPGAN/code/train/base_trainer.py:413
↓ 1 callersMethodget_rewards
(self, scores, baseline)
Projects/UPGAN/code/train/trainer.py:53
↓ 1 callersFunctioninit_model
(args, user_total, item_total, entity_total, r
Projects/UPGAN/code/pretrain/init.py:6
↓ 1 callersFunctioninit_model
(args, user_total, item_total, entity_total, r
Projects/UPGAN/code/train/init.py:7
↓ 1 callersMethodlabel_triples
(self, hop_limit)
Projects/UPGAN/code/pretrain/base_trainer.py:81
↓ 1 callersMethodlabel_triples
(self, hop_limit)
Projects/UPGAN/code/train/base_trainer.py:81
↓ 1 callersFunctionload_i2kg_map
(filename, i_map, e_map)
Projects/UPGAN/code/util/utils.py:151
↓ 1 callersMethodload_pretrain
(self, args)
Projects/UPGAN/code/pretrain/base_trainer.py:284
↓ 1 callersMethodload_pretrain
(self, args)
Projects/UPGAN/code/train/base_trainer.py:286
↓ 1 callersMethodloss_def
(self)
Projects/UPGAN/code/pretrain/trainer.py:46
↓ 1 callersMethodloss_def
(self)
Projects/UPGAN/code/train/trainer.py:48
↓ 1 callersFunctionmain
()
Projects/UPGAN/code/main_upgan.py:140
↓ 1 callersFunctionmain
()
Projects/UPGAN/code/main_pretrain.py:136
↓ 1 callersFunctionmake_kg_graph_batch
(path_list, max_hop, ent_hop, hop_map)
Projects/UPGAN/code/util/triple_kernel.py:42
↓ 1 callersFunctionmake_triple_graph_rs_batch
(path_list, batch_ents, ent_map, interval=100000)
Projects/UPGAN/code/util/triple_kernel.py:10
↓ 1 callersMethodmodel_def
(self)
Projects/UPGAN/code/pretrain/trainer.py:36
↓ 1 callersMethodmodel_def
(self)
Projects/UPGAN/code/train/trainer.py:38
↓ 1 callersMethodmodule_def
(self, args)
Projects/UPGAN/code/Model/UGAT.py:40
↓ 1 callersMethodmodule_def
(self, args)
Projects/UPGAN/code/Model/UGAT_mlp.py:41
↓ 1 callersMethodoptim_def
(self, lr)
Projects/UPGAN/code/pretrain/trainer.py:29
↓ 1 callersMethodoptim_def
(self, lr)
Projects/UPGAN/code/train/trainer.py:30
↓ 1 callersMethodpop_prior
(self, neg_tails)
Projects/UPGAN/code/train/trainer.py:200
↓ 1 callersMethodpop_prior_def
(self)
Projects/UPGAN/code/train/trainer.py:72
↓ 1 callersMethodpretrain_G
(self)
Projects/UPGAN/code/train/trainer.py:92
↓ 1 callersMethodpretrain_epoch_G
(self)
Projects/UPGAN/code/train/trainer.py:278
↓ 1 callersMethodrandom_prior
(self, neg_tails)
Projects/UPGAN/code/train/trainer.py:205
↓ 1 callersMethodregularization_loss
计算张量范数 :param weight_list: :param p: 范数计算中的幂指数值,默认求2范数 :param weight_decay: :return:
Projects/UPGAN/code/util/Regularization.py:54
↓ 1 callersMethodrel_update
(self, B, tail_rep, rel, edge_list, weight_list)
Projects/UPGAN/code/Model/layers.py:38
↓ 1 callersMethodremap_entity_batch
(self, max_hop, ent_hop)
Projects/UPGAN/code/pretrain/base_trainer.py:423
↓ 1 callersMethodremap_entity_batch
(self, max_hop, ent_hop)
Projects/UPGAN/code/train/base_trainer.py:425
↓ 1 callersMethodsample_all_reverse
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:396
↓ 1 callersMethodsample_all_reverse
(self)
Projects/UPGAN/code/train/base_trainer.py:398
↓ 1 callersMethodsample_epoch_edges
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:348
↓ 1 callersMethodsample_epoch_edges
(self)
Projects/UPGAN/code/train/base_trainer.py:350
↓ 1 callersMethodsample_epoch_reverse
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:372
↓ 1 callersMethodsample_epoch_reverse
(self)
Projects/UPGAN/code/train/base_trainer.py:374
↓ 1 callersMethodtrain
(self, start_epoch, end_epoch)
Projects/UPGAN/code/pretrain/trainer.py:60
↓ 1 callersMethodtrain_epoch_kg
(self)
Projects/UPGAN/code/pretrain/trainer.py:133
↓ 1 callersMethodtrain_epoch_kg
(self)
Projects/UPGAN/code/train/trainer.py:321
↓ 1 callersFunctionuniverse_sample
(user_dict, entity_dict, entity_tot, relation_tot, num_sample=10, sample_flag=True)
Projects/UPGAN/code/util/sage_kernel.py:23
↓ 1 callersMethodweight_info
打印权重列表信息 :param weight_list: :return:
Projects/UPGAN/code/util/Regularization.py:74
↓ 1 callersFunctionwrite_res
(output, res_dict, relation_num)
Projects/UPGAN/code/train/evaluation.py:26
Method__getitem__
(self, idx)
Projects/UPGAN/code/util/dataset.py:130
Method__init__
(self, entity_total, relation_total, args)
Projects/UPGAN/code/Model/dot_2layer.py:18
Method__init__
(self, user_total, linked_total, entity_total, relation_total, args)
Projects/UPGAN/code/Model/UGAT.py:20
Method__init__
(self, in_features, out_features, match_matrix, devidce)
Projects/UPGAN/code/Model/layers.py:13
Method__init__
(self)
Projects/UPGAN/code/Model/base_model.py:9
Method__init__
(self, entity_total, relation_total, args)
Projects/UPGAN/code/Model/concat_1layer.py:18
Method__init__
(self, entity_total, relation_total, args)
Projects/UPGAN/code/Model/DistMult.py:16
Method__init__
(self, entity_total, relation_total, args)
Projects/UPGAN/code/Model/generator_concat.py:18
Method__init__
(self, entity_total, relation_total, args)
Projects/UPGAN/code/Model/concat_2layer.py:18
Method__init__
(self, entity_total, relation_total, args)
Projects/UPGAN/code/Model/generator.py:18
Method__init__
(self, user_total, linked_total, entity_total, relation_total, args)
Projects/UPGAN/code/Model/UGAT_mlp.py:20
Method__init__
(self, args, logger=None)
Projects/UPGAN/code/pretrain/trainer.py:17
Method__init__
(self, model, use_cuda, middle_file, logger, relation_num, k=10, e_map=None)
Projects/UPGAN/code/train/evaluation.py:50
Method__init__
(self, args, logger=None)
Projects/UPGAN/code/train/trainer.py:17
Method__init__
(self, triple_list, entity_num, relation_num, args)
Projects/UPGAN/code/util/dataset.py:77
Method__init__
:param model 模型 :param weight_decay:正则化参数 :param p: 范数计算中的幂指数值,默认求2范数, 当p=0为L2正则化,p=1为L1正则化
Projects/UPGAN/code/util/Regularization.py:9
Method__len__
(self)
Projects/UPGAN/code/util/dataset.py:94
Method__repr__
(self)
Projects/UPGAN/code/Model/layers.py:59
Method__repr__
(self)
Projects/UPGAN/code/Model/layers.py:190
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/dot_2layer.py:10
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/UGAT.py:10
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/concat_1layer.py:10
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/DistMult.py:8
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/generator_concat.py:10
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/concat_2layer.py:10
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/generator.py:10
Functionbuild_model
(args, user_total, item_total, entity_total, relation_total, i_map=None, e_map=None, new_map=None, share_total
Projects/UPGAN/code/Model/UGAT_mlp.py:10
Methodcalc_attn_kg
(self, head_rep, tail_rep, edge_list, requires_grad=False)
Projects/UPGAN/code/Model/layers.py:104
Methodcalc_attn_rs
(self, head_rep, tail_rep, edge_list, rs_rel)
Projects/UPGAN/code/Model/layers.py:114
Functioncalc_bern
(triple_list)
Projects/UPGAN/code/util/dataset.py:31
Functionchange_head
(triple, entityTotal, relationTotal, train_hr_t)
Projects/UPGAN/code/util/dataset.py:7
Functionchange_tail
(triple, entityTotal, relationTotal, train_hr_t)
Projects/UPGAN/code/util/dataset.py:19
Methoddump_emb
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:235
Methoddump_emb
(self)
Projects/UPGAN/code/train/base_trainer.py:235
Methodencode_kg
(self, ent_emb, rel_emb)
Projects/UPGAN/code/Model/base_model.py:20
Methodencode_user
(self, rels, e1_embedded_user)
Projects/UPGAN/code/Model/UGAT.py:162
Methodencode_user
(self, rels, e1_embedded_user)
Projects/UPGAN/code/Model/UGAT_mlp.py:163
Methodeval_sample_diversity
(self, epoch=0)
Projects/UPGAN/code/pretrain/base_trainer.py:187
Methodeval_sample_popularity
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:115
Methodeval_sample_popularity
(self)
Projects/UPGAN/code/train/base_trainer.py:115
Methodeval_sample_quality
(self)
Projects/UPGAN/code/pretrain/base_trainer.py:154
← previousnext →101–200 of 275, ranked by callers