Functions — github.com/RUCDM/KB4Rec

hub / github.com/RUCDM/KB4Rec / functions

Functions275 in github.com/RUCDM/KB4Rec

⨍ Functions 275◇ Types & classes 18

Methodevaluate

(self, e1, rel, e1_embedded_all, pretrain=False)

Projects/UPGAN/code/Model/dot_2layer.py:91

Methodevaluate

(self, heads, rels, batch_rel_cand)

Projects/UPGAN/code/Model/base_model.py:26

Methodevaluate

(self, e1, rel, e1_embedded_all, pretrain=False)

Projects/UPGAN/code/Model/concat_1layer.py:90

Methodevaluate

(self, e1, rel, e1_embedded_all)

Projects/UPGAN/code/Model/DistMult.py:54

Methodevaluate

(self, e1, rel, e1_embedded_all, pretrain=False)

Projects/UPGAN/code/Model/generator_concat.py:88

Methodevaluate

(self, e1, rel, e1_embedded_all, pretrain=False)

Projects/UPGAN/code/Model/concat_2layer.py:91

Methodevaluate

(self, e1, rel, e1_embedded_all, pretrain=False)

Projects/UPGAN/code/Model/generator.py:90

Methodevaluate

(self, heads, rels, batch_rel_cand, pretrain=False)

Projects/UPGAN/code/Model/UGAT_mlp.py:102

Methodevaluate_single

(self, filename)

Projects/UPGAN/code/pretrain/base_trainer.py:242

Methodform_query

(self, e1, rel)

Projects/UPGAN/code/Model/dot_2layer.py:48

Methodform_query

(self, heads, rels, e1_embedded_user)

Projects/UPGAN/code/Model/base_model.py:46

Methodform_query

(self, e1, rel)

Projects/UPGAN/code/Model/concat_1layer.py:47

Methodform_query

(self, e1, rel)

Projects/UPGAN/code/Model/generator_concat.py:47

Methodform_query

(self, e1, rel)

Projects/UPGAN/code/Model/concat_2layer.py:48

Methodform_query

(self, e1, rel)

Projects/UPGAN/code/Model/generator.py:47

Methodform_query

(self, heads, rels, e1_embedded_user, pretrain=False)

Projects/UPGAN/code/Model/UGAT_mlp.py:124

Methodforward

(self, heads, rels, tails, e1_embedded_user, pretrain=False)

Projects/UPGAN/code/Model/UGAT.py:71

Methodforward

(self, input, tail_rep, rel_Adj, activate=True)

Projects/UPGAN/code/Model/layers.py:25

Methodforward

(self, heads, rels, tails, e1_embedded_user)

Projects/UPGAN/code/Model/base_model.py:23

Methodforward

(self, heads, rels, tails, e1_embedded_user, pretrain=False)

Projects/UPGAN/code/Model/UGAT_mlp.py:72

Methodforward

(self, model)

Projects/UPGAN/code/util/Regularization.py:36

Methodforward_batch

:param heads: (B,) :param rels: (B,) :param tails: (B.) :param e1_embedded_user:(B, emb) :return:

Projects/UPGAN/code/Model/DistMult.py:82

Methodforward_kg

(self, e1, rel, all_e_ids=None)

Projects/UPGAN/code/Model/dot_2layer.py:128

Methodforward_kg

(self, head_rep, tail_rep, edge_list, requires_grad=False)

Projects/UPGAN/code/Model/layers.py:152

Methodforward_kg

(self, e1, rel, all_e_ids=None)

Projects/UPGAN/code/Model/concat_1layer.py:127

Methodforward_kg

(self, e1, rel, all_e_ids=None)

Projects/UPGAN/code/Model/generator_concat.py:124

Methodforward_kg

(self, e1, rel, all_e_ids=None)

Projects/UPGAN/code/Model/concat_2layer.py:128

Methodforward_kg

(self, e1, rel, all_e_ids=None)

Projects/UPGAN/code/Model/generator.py:127

Methodforward_triple

:param heads: (B, ) :param rels: (B, ) :param tails: (B, N) :return:

Projects/UPGAN/code/Model/dot_2layer.py:103

Methodforward_triple

:param heads: (B, ) :param rels: (B, ) :param tails: (B, N) :return:

Projects/UPGAN/code/Model/concat_1layer.py:102

Methodforward_triple

:param heads: (B, ) :param rels: (B, ) :param tails: (B, N) :return:

Projects/UPGAN/code/Model/generator_concat.py:100

Methodforward_triple

:param heads: (B, ) :param rels: (B, ) :param tails: (B, N) :return:

Projects/UPGAN/code/Model/concat_2layer.py:103

Methodforward_triple

:param heads: (B, ) :param rels: (B, ) :param tails: (B, N) :return:

Projects/UPGAN/code/Model/generator.py:102

Methodfreeze_part

(self)

Projects/UPGAN/code/Model/UGAT.py:168

Methodfreeze_part

(self)

Projects/UPGAN/code/Model/UGAT_mlp.py:169

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/dot_2layer.py:88

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/base_model.py:84

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/concat_1layer.py:87

Methodget_candidates

(self)

Projects/UPGAN/code/Model/DistMult.py:51

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/generator_concat.py:85

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/concat_2layer.py:88

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/generator.py:87

Methodget_candidates

(self, pretrain=False)

Projects/UPGAN/code/Model/UGAT_mlp.py:93

Methodget_embeddings

(self, ents)

Projects/UPGAN/code/Model/UGAT_mlp.py:99

Methodget_norm

(self)

Projects/UPGAN/code/Model/base_model.py:38

Methodget_top

(self, eval_loader, prefix="./", topk=100)

Projects/UPGAN/code/train/evaluation.py:59

Methodget_weight

(self)

Projects/UPGAN/code/Model/layers.py:22

Methodl2_loss

(vector)

Projects/UPGAN/code/Model/base_model.py:30

Methodl3_loss

(vector)

Projects/UPGAN/code/Model/base_model.py:34

Methodload_G_pretrain

(self)

Projects/UPGAN/code/pretrain/base_trainer.py:278

Methodload_G_pretrain

(self)

Projects/UPGAN/code/train/base_trainer.py:280

Functionmake_RGCN_kernel_batch

(reserved_ents, path_reserve, entity_tot, relation_tot)

Projects/UPGAN/code/util/sage_kernel.py:4

Functionmake_RGCN_kernel_sample

(user_dict, entity_dict, entity_tot, relation_tot, num_sample=10, sample_flag=True)

Projects/UPGAN/code/util/sage_kernel.py:17

Functionmake_query_graph_hop_batch

(path_list, max_hop, ent_hop, hop_map, interval=100000)

Projects/UPGAN/code/util/triple_kernel.py:106

Functionmake_triple_graph_kg_single_hop

(path_list, batch_ents, ent_map_now, ent_map_pre, interval=100000)

Projects/UPGAN/code/util/triple_kernel.py:69

Methodprediction

(self, query_now)

Projects/UPGAN/code/Model/UGAT.py:146

Methodprediction

(self, query_now)

Projects/UPGAN/code/Model/UGAT_mlp.py:148

Functionprint_rank_list

(raw_list, fil_list, temp_str, logger)

Projects/UPGAN/code/util/utils.py:131

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/dot_2layer.py:61

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/base_model.py:49

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/concat_1layer.py:60

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/generator_concat.py:59

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/concat_2layer.py:61

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/generator.py:60

Methodquery_judge

(self, query_now, tails)

Projects/UPGAN/code/Model/UGAT_mlp.py:153

Methodrandom_sample

(self, heads, rels, neg_tails, n_sample=1, temperature=1.0)

Projects/UPGAN/code/train/trainer.py:227

Methodsample_all_eges

(self)

Projects/UPGAN/code/pretrain/base_trainer.py:335

Methodsample_all_eges

(self)

Projects/UPGAN/code/train/base_trainer.py:337

Methodsave_ckpt

(self)

Projects/UPGAN/code/pretrain/base_trainer.py:262

Methodsave_ckpt

(self)

Projects/UPGAN/code/train/base_trainer.py:262

Methodunfreeze_part

(self)

Projects/UPGAN/code/Model/UGAT.py:174

Methodunfreeze_part

(self)

Projects/UPGAN/code/Model/UGAT_mlp.py:175

Functionwrite_res

(output, res_dict, relation_num)

Projects/UPGAN/code/util/utils.py:100

Functionwrite_res_inductive

(output, res_dict)

Projects/UPGAN/code/train/evaluation.py:39

Functionwrite_res_inductive

(output, res_dict)

Projects/UPGAN/code/util/utils.py:113

← previous201–275 of 275, ranked by callers