MCPcopy Create free account

hub / github.com/ZZy979/pytorch-tutorial / functions

Functions479 in github.com/ZZy979/pytorch-tutorial

Methodforward
:param gs: List[DGLGraph] 基于每条元路径的邻居组成的图 :param node_feat: tensor(N, d_in) 输入顶点特征,N为给定类型的顶点个数 :param edge_feat_name: str 元路径实
gnn/magnn/model.py:128
Methodforward
:param gs: Dict[str, List[DGLGraph]] 顶点类型到其对应的基于每条元路径的邻居组成的图的映射 :param node_feats: Dict[str, tensor(N_i, d_in)] 顶点类型到输入顶点特征的映射,N_i为对应
gnn/magnn/model.py:166
Methodforward
:param blocks: List[DGLBlock] 目标顶点类型对应的每条元路径的邻居组成的图(block) :param node_feats: Dict[str, tensor(N_i, d_in)] 所有顶点类型到输入顶点特征的映射,N_i为对应类型的
gnn/magnn/model.py:212
Methodforward
:param gs: Dict[str, List[DGLGraph]] 顶点类型到其对应的基于每条元路径的邻居组成的图的映射 :param node_feats: Dict[str, tensor(N_i, d_in)] 顶点类型到输入顶点特征的映射,N_i为对应
gnn/magnn/model.py:253
Methodforward
:param g: DGLGraph 邻居-目标顶点二分图 :param feat_src: tensor(N_src, d) 邻居顶点输入特征 :param feat_dst: tensor(N_dst, d) 目标顶点输入特征 :
gnn/heco/model.py:37
Methodforward
:param h: tensor(N, M, d) 顶点基于不同元路径/类型的嵌入,N为顶点数,M为元路径/类型数 :return: tensor(N, d) 顶点的最终嵌入
gnn/heco/model.py:83
Methodforward
:param bgs: List[DGLGraph] 各类型邻居到目标顶点的二分图 :param feats: List[tensor(N_i, d)] 输入顶点特征,feats[0]为目标顶点特征,feats[i]对应bgs[i-1] :retur
gnn/heco/model.py:114
Methodforward
:param mgs: List[DGLGraph] 基于元路径的邻居图 :param feat: tensor(N, d) 输入顶点特征 :return: tensor(N, d) 输出顶点特征
gnn/heco/model.py:146
Methodforward
:param z_sc: tensor(N, d) 目标顶点在网络结构视图下的嵌入 :param z_mp: tensor(N, d) 目标顶点在元路径视图下的嵌入 :param pos: tensor(N, N) 0-1张量,每个顶点的正样本
gnn/heco/model.py:196
Methodforward
:param bgs: List[DGLGraph] 各类型邻居到目标顶点的二分图 :param mgs: List[DGLGraph] 基于元路径的邻居图 :param feats: List[tensor(N_i, d_in)] 输入顶点特征,f
gnn/heco/model.py:245
Methodforward
:param g: DGLGraph :param x: tensor(N, d_in) 输入顶点特征,N为g的顶点数 :return: tensor(N, d_out) 输出顶点特征
gnn/gcn/model.py:18
Methodhas_cache
(self)
gnn/data/acm.py:120
Methodhas_cache
(self)
gnn/data/acm.py:206
Methodhas_cache
(self)
gnn/data/dblp.py:176
Methodhas_cache
(self)
gnn/data/dblp.py:262
Methodhas_cache
(self)
gnn/data/aminer.py:123
Methodhas_cache
(self)
gnn/data/aminer.py:261
Methodhas_cache
(self)
gnn/data/heco.py:106
Methodhas_cache
(self)
gnn/data/imdb.py:135
Methodhas_cache
(self)
gnn/data/imdb.py:222
Functionhits_at
计算Hits@n = #(rank_i <= n) / N 例如,predicts=[[2, 0, 1], [2, 1, 0], [1, 0, 2]], answers=[1, 2, 0], ranks=[3, 1, 2], Hits@2=2/3 :param n: int 要计
gnn/utils/metrics.py:61
Methodinference
离线推断所有顶点的最终嵌入(不使用邻居采样) :param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in_i)] 顶点类型到输入顶点特征的映射 :param device: torc
gnn/rhgnn/model.py:374
Methodload
(self)
gnn/data/acm.py:177
Methodload
(self)
gnn/data/dblp.py:64
Methodload
(self)
gnn/data/dblp.py:236
Methodload
(self)
gnn/data/aminer.py:65
Methodload
(self)
gnn/data/aminer.py:185
Methodload
(self)
gnn/data/imdb.py:61
Methodload
(self)
gnn/data/imdb.py:194
Functionmean_reciprocal_rank
计算平均倒数排名(MRR) = sum(1 / rank_i) / N,如果答案不在预测结果中,则排名按∞计算 例如,predicts=[[2, 0, 1], [2, 1, 0], [1, 0, 2]], answers=[1, 2, 8], ranks=[3, 2, ∞], MR
gnn/utils/metrics.py:47
Methodmetapaths
(self)
gnn/data/acm.py:136
Methodmetapaths
(self)
gnn/data/dblp.py:192
Methodmetapaths
(self)
gnn/data/heco.py:122
Methodmetapaths
(self)
gnn/data/heco.py:160
Methodmetapaths
(self)
gnn/data/heco.py:206
Methodmetapaths
(self)
gnn/data/heco.py:235
Methodmetapaths
(self)
gnn/data/heco.py:264
Methodmetapaths
(self)
gnn/data/imdb.py:151
Functionnmi_ari_score
计算NMI和ARI得分 :param logits: tensor(N, d) 预测概率,N为样本数 :param labels: tensor(N) 正确标签 :return: float, float NMI和ARI得分
gnn/utils/metrics.py:32
Methodnum_classes
(self)
gnn/data/acm.py:132
Methodnum_classes
(self)
gnn/data/acm.py:218
Methodnum_classes
(self)
gnn/data/dblp.py:188
Methodnum_classes
(self)
gnn/data/dblp.py:274
Methodnum_classes
(self)
gnn/data/aminer.py:136
Methodnum_classes
(self)
gnn/data/aminer.py:273
Methodnum_classes
(self)
gnn/data/heco.py:118
Methodnum_classes
(self)
gnn/data/imdb.py:147
Methodnum_classes
(self)
gnn/data/imdb.py:234
Methodpos
(self)
gnn/data/heco.py:130
Methodpredict_ntype
(self)
gnn/data/acm.py:140
Methodpredict_ntype
(self)
gnn/data/dblp.py:196
Methodpredict_ntype
(self)
gnn/data/heco.py:126
Methodpredict_ntype
(self)
gnn/data/heco.py:164
Methodpredict_ntype
(self)
gnn/data/heco.py:210
Methodpredict_ntype
(self)
gnn/data/heco.py:239
Methodpredict_ntype
(self)
gnn/data/heco.py:268
Methodpredict_ntype
(self)
gnn/data/imdb.py:155
Methodprocess
(self)
gnn/dgl/graph_clf_hetero.py:43
Methodprocess
(self)
gnn/data/acm.py:63
Methodprocess
(self)
gnn/data/acm.py:183
Methodprocess
(self)
gnn/data/dblp.py:70
Methodprocess
(self)
gnn/data/dblp.py:242
Methodprocess
(self)
gnn/data/dgl.py:82
Methodprocess
(self)
gnn/data/dgl.py:112
Methodprocess
(self)
gnn/data/aminer.py:73
Methodprocess
(self)
gnn/data/aminer.py:189
Methodprocess
(self)
gnn/data/heco.py:65
Methodprocess
(self)
gnn/data/imdb.py:67
Methodprocess
(self)
gnn/data/imdb.py:200
Methodsave
(self)
gnn/data/acm.py:174
Methodsave
(self)
gnn/data/dblp.py:61
Methodsave
(self)
gnn/data/dblp.py:233
Methodsave
(self)
gnn/data/aminer.py:57
Methodsave
(self)
gnn/data/aminer.py:182
Methodsave
(self)
gnn/data/heco.py:51
Methodsave
(self)
gnn/data/imdb.py:58
Methodsave
(self)
gnn/data/imdb.py:191
Functionshow_random_image
(trainloader)
beginner/cifar10_tutorial.py:23
Functionstack_reducer
(nodes)
gnn/hetgnn/model.py:141
← previous401–479 of 479, ranked by callers