MCPcopy Create free account

hub / github.com/THUDM/HGB / functions

Functions2,023 in github.com/THUDM/HGB

⨍ Functions 2,023◇ Types & classes 293
↓ 2 callersFunctiondcg_at_k
Score is discounted cumulative gain (dcg) Relevance is positive real values. Can use binary as the previous methods. Returns: Dis
Recom/KGAT/Model/utility/metrics.py:43
↓ 2 callersFunctiondelete_files
(files_)
NC/benchmark/scripts/NC_F1.py:96
↓ 2 callersFunctiondelete_files
(files_)
LP/benchmark/scripts/LP_AUC_MRR.py:124
↓ 2 callersFunctiondense_tensor_to_sparse
(dense_mx)
TC/HGAT/model/code/utils.py:244
↓ 2 callersFunctiondot
Wrapper for tf.matmul (sparse vs dense).
Recom/KGCN/src/gcn/layers.py:5
↓ 2 callersFunctiondot
Wrapper for tf.matmul (sparse vs dense).
Recom/KGNN-LS/src/gcn/layers.py:5
↓ 2 callersMethodencode
(self, data)
LP/RGCN/GNN.py:54
↓ 2 callersMethodencode
(self, data)
LP/HetGNN/code/homoGNN.py:65
↓ 2 callersFunctionensureDir
(dir_path)
Recom/baseline/Model/utility/helper.py:15
↓ 2 callersFunctionevaluate
(model, g, features, labels, mask, loss_func)
NC/benchmark/methods/HAN/main.py:21
↓ 2 callersFunctionevaluate
(model, g, features, labels, mask, loss_func)
NC/HAN/main.py:20
↓ 2 callersFunctionevaluate
(model, data, edge_index, labels)
LP/HetGNN/code/homoGNN.py:519
↓ 2 callersFunctionevaluate
(out_feat, pos_num)
LP/GATNE/src/homGNN.py:237
↓ 2 callersMethodevaluate
(self, pred)
NC/benchmark/methods/HetSANN/HetSANN_MRV/scripts/data_loader.py:133
↓ 2 callersFunctionevaluate_results_nc
(embeddings, labels, num_classes)
LP/benchmark/methods/MAGNN/utils/tools.py:51
↓ 2 callersFunctionevaluate_results_nc
(embeddings, labels, num_classes)
LP/benchmark/methods/MAGNN_ini/utils/tools.py:51
↓ 2 callersFunctionfile2list
(file_name, eval_type)
LP/GATNE/src/homGNN.py:257
↓ 2 callersFunctionfiles_exist
(files)
NC/benchmark/methods/RSHN/torch_geometric/data/dataset.py:15
↓ 2 callersFunctionfiles_exist
(files)
NC/RSHN/torch_geometric/data/dataset.py:15
↓ 2 callersMethodgcn_conv
(self,Xs,H)
NC/benchmark/methods/GTN/model.py:40
↓ 2 callersMethodgcn_conv
(self,X,H)
NC/GTN/model.py:39
↓ 2 callersFunctiongen_data
(embed, edge_list)
LP/HetGNN/code/homoGNN.py:438
↓ 2 callersFunctiongen_edge_list
(args, edge_data_)
LP/HetGNN/code/homoGNN.py:350
↓ 2 callersFunctiongen_embed
(args, with_paper=True)
LP/HetGNN/code/homoGNN.py:312
↓ 2 callersFunctiongenerate
(network_data, num_walks, walk_length, schema, file_name, window_size, num_workers, walk_file)
LP/GATNE/src/utils.py:221
↓ 2 callersFunctiongenerate_neighbors
(network_data, vocab, num_nodes, edge_types, neighbor_samples)
LP/GATNE/src/utils.py:232
↓ 2 callersMethodgenerate_test_feed_dict
(self, model, user_batch, item_batch, drop_flag=True)
Recom/KGAT/Model/utility/loader_nfm.py:102
↓ 2 callersMethodget
(self, idx)
NC/benchmark/methods/RSHN/torch_geometric/data/in_memory_dataset.py:64
↓ 2 callersMethodget
(self, idx)
NC/RSHN/torch_geometric/data/in_memory_dataset.py:64
↓ 2 callersMethodget_edge_type
(self, info)
NC/benchmark/scripts/data_loader.py:230
↓ 2 callersMethodget_edge_type
(self, info)
NC/benchmark/methods/HGT/data_loader.py:183
↓ 2 callersMethodget_edge_type
(self)
NC/benchmark/methods/HetSANN/HetSANN_MRV/find_meta.py:86
↓ 2 callersMethodget_edge_type
(self, info)
NC/benchmark/methods/HetSANN/HetSANN_MRV/scripts/data_loader.py:181
↓ 2 callersMethodget_edge_type
(self, info)
NC/benchmark/methods/RGCN/scripts/data_loader.py:125
↓ 2 callersMethodget_edge_type
(self, info)
LP/RGCN-WN18/code/scripts/data_loader.py:223
↓ 2 callersMethodget_edge_type
(self, info)
LP/benchmark/scripts/data_loader.py:276
↓ 2 callersFunctionget_feed_dict
(model, data, start, end, is_train)
Recom/KGCN/src/train.py:79
↓ 2 callersFunctionget_feed_dict
(model, data, start, end, is_train)
Recom/KGNN-LS/src/train.py:95
↓ 2 callersMethodget_loss
(self, embed, triplets, labels)
LP/RGCN-WN18/code/link_predict.py:55
↓ 2 callersMethodget_loss
(self, embed, triplets, labels)
LP/benchmark/methods/RGCN/link_predict.py:60
↓ 2 callersMethodget_node_type
(self, node_id)
NC/benchmark/methods/HGT/data_loader.py:178
↓ 2 callersMethodget_node_type
(self, node_id)
NC/benchmark/methods/HetSANN/HetSANN_MRV/scripts/data_loader.py:176
↓ 2 callersMethodget_node_type
(self, node_id)
NC/benchmark/methods/RGCN/scripts/data_loader.py:120
↓ 2 callersMethodget_node_type
(self, node_id)
LP/benchmark/scripts/data_loader.py:271
↓ 2 callersMethodget_node_type
(self, node_id)
LP/benchmark/methods/HGT/data_loader.py:238
↓ 2 callersFunctionget_score
(local_model, node1, node2)
LP/benchmark/methods/GATNE/utils.py:261
↓ 2 callersFunctionget_score
(local_model, node1, node2)
LP/GATNE/src/utils.py:252
↓ 2 callersMethodget_scores
(self, sess, feed_dict)
Recom/KGCN/src/model.py:203
↓ 2 callersMethodget_scores
(self, sess, feed_dict)
Recom/KGNN-LS/src/model.py:201
↓ 2 callersMethodget_test_neigh
(self)
LP/benchmark/scripts/data_loader.py:346
↓ 2 callersMethodget_test_neigh_w_random
(self)
LP/RGCN-WN18/code/scripts/data_loader.py:371
↓ 2 callersFunctionget_user_record
(data, is_train)
Recom/KGCN/src/train.py:144
↓ 2 callersFunctionget_user_record
(data, is_train)
Recom/KGNN-LS/src/train.py:175
↓ 2 callersMethodlist_to_sp_mat
(self, li)
LP/RGCN-WN18/code/scripts/data_loader.py:238
↓ 2 callersFunctionload_DBLP_data
(prefix='data/preprocessed/DBLP_processed')
NC/MAGNN/utils/data.py:36
↓ 2 callersFunctionload_LastFM_data
(prefix='data/preprocessed/LastFM_processed')
LP/benchmark/methods/MAGNN_ini/utils/data.py:79
↓ 2 callersMethodload_checkpoint
Load the latest checkpoint.
NC/benchmark/methods/HAN/utils.py:384
↓ 2 callersFunctionload_data
(data_file_name, n_features, n_samples)
NC/HetGNN/code/link_prediction_model.py:28
↓ 2 callersFunctionload_data
(data_file_name, n_features, n_samples)
NC/HetGNN/code/node_classification_model.py:28
↓ 2 callersFunctionload_data
(prefix='DBLP')
NC/benchmark/methods/HGT/utils/data.py:7
↓ 2 callersFunctionload_data
(dataset, feat_type=0)
NC/benchmark/methods/HAN/utils.py:337
↓ 2 callersFunctionload_data
(prefix='DBLP')
NC/benchmark/methods/GNN/utils/data.py:7
↓ 2 callersFunctionload_data
(data_file_name, n_features, n_samples)
LP/HetGNN/code/link_prediction_model.py:28
↓ 2 callersFunctionload_data
(data_file_name, n_features, n_samples)
LP/HetGNN/code/node_classification_model.py:28
↓ 2 callersFunctionload_data
(prefix='DBLP')
LP/benchmark/methods/baseline/utils/data.py:7
↓ 2 callersFunctionload_data
(path="../data/citeseer/", dataset="citeseer")
TC/HGAT/model/code/utils.py:9
↓ 2 callersFunctionload_feature_data
(f_name)
LP/GATNE/src/utils.py:134
↓ 2 callersMethodload_labels
return labels dict num_classes: total number of labels total: total number of labeled data count: number
NC/benchmark/scripts/NC_F1.py:28
↓ 2 callersMethodload_labels
return labels dict num_classes: total number of labels total: total number of labeled data count: number
NC/benchmark/scripts/data_loader.py:192
↓ 2 callersMethodload_labels
return labels dict num_classes: total number of labels total: total number of labeled data count: number
NC/benchmark/methods/HGT/data_loader.py:145
↓ 2 callersMethodload_labels
return labels dict num_classes: total number of labels total: total number of labeled data count: number
NC/benchmark/methods/HetSANN/HetSANN_MRV/scripts/data_loader.py:143
↓ 2 callersMethodload_labels
return labels dict num_classes: total number of labels total: total number of labeled data count: number
NC/benchmark/methods/RGCN/scripts/data_loader.py:87
↓ 2 callersMethodload_links
return links dict total: total number of links count: a dict of int, number of links for each type data:
LP/benchmark/scripts/LP_AUC_MRR.py:31
↓ 2 callersMethodload_links
return links dict total: total number of links count: a dict of int, number of links for each type meta:
LP/benchmark/scripts/data_loader.py:516
↓ 2 callersMethodload_links
return links dict total: total number of links count: a dict of int, number of links for each type meta:
LP/benchmark/methods/HGT/data_loader.py:482
↓ 2 callersFunctionload_test_data
(dl, random_test=True)
LP/benchmark/methods/GATNE/utils.py:119
↓ 2 callersFunctionload_training_data
(f_name)
LP/GATNE/src/utils.py:86
↓ 2 callersFunctionmodel
()
NC/benchmark/methods/HetGNN/code/DBLP/do_class.py:51
↓ 2 callersFunctionmodel
()
NC/benchmark/methods/HetGNN/code/IMDB/do_class.py:62
↓ 2 callersMethodnode_list
(self, nodes, num_walks)
LP/benchmark/methods/GATNE/walk.py:40
↓ 2 callersMethodnode_list
(self, nodes, num_walks)
LP/GATNE/src/walk.py:40
↓ 2 callersFunctionnode_norm_to_edge_norm
(g, node_norm)
LP/RGCN/link_predict.py:82
↓ 2 callersMethodp_content_agg
(self, id_batch)
NC/HetGNN/code/tools.py:80
↓ 2 callersMethodp_content_agg
(self, id_batch)
NC/benchmark/methods/HetGNN/code/DBLP/tools.py:83
↓ 2 callersMethodp_content_agg
(self, id_batch)
LP/HetGNN/code/tools.py:80
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
NC/MAGNN/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
NC/benchmark/methods/MAGNN/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
NC/benchmark/methods/HGT/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
NC/benchmark/methods/baseline/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
LP/MAGNN/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
LP/benchmark/methods/MAGNN/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
LP/benchmark/methods/HGT/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
LP/benchmark/methods/MAGNN_ini/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
LP/benchmark/methods/baseline/utils/tools.py:129
↓ 2 callersFunctionparse_adjlist_LastFM
(adjlist, edge_metapath_indices, samples=None, exclude=None, offset=None, mode=None)
TC/HGAT/model/code/baseline/utils/tools.py:129
↓ 2 callersFunctionparse_args
()
LP/GATNE/src/utils.py:22
↓ 2 callersFunctionparse_args
()
Recom/baseline/Model/utility/parser.py:9
↓ 2 callersFunctionparse_args
()
Recom/KGAT/Model/utility/parser.py:9
↓ 2 callersFunctionperturb_and_get_raw_rank
Perturb one element in the triplets
LP/RGCN/utils.py:193
↓ 2 callersFunctionpreprocess_adj_bias
(adj, norm=False)
Recom/KGCN/src/model.py:18
← previousnext →201–300 of 2,023, ranked by callers