![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
小黑图神经网络探索之旅
爱喝喜茶爱吃烤冷面的小黑黑
这个作者很懒,什么都没留下…
展开
-
小黑demo:transH
import torchimport torch.nn as nnimport torch.nn.functional as Fclass TransH(nn.Module): def __init__(self,ent_tot,rel_tot,dim = 100,p_norm = 1,norm_flag = True,margin = None,epsilon = None): super(TransH,self).__init__() self..原创 2022-02-04 16:15:13 · 1144 阅读 · 1 评论 -
node2vec之小黑尝试
参数设定import warningsimport randomwarnings.filterwarnings('ignore')import argparseimport numpy as npimport networkx as nx#import node2vecfrom gensim.models import Word2Vecimport randomnp.random.seed(1)def parse_args(): ''' Parses the node2原创 2022-01-10 14:54:01 · 403 阅读 · 0 评论 -
小黑demo:TransD模型
import torchimport torch.nn as nnimport torch.nn.functional as Fclass TransD(nn.Module): def __init__(self,ent_tot,rel_tot,dim_e = 100,dim_r = 100,p_norm = 1,norm_flag = True,margin = None,epsilon = None): super(TransD,self).__init__() ...原创 2022-02-05 16:23:44 · 2145 阅读 · 2 评论 -
图神经网络学习笔记:Metapath2vec
1.图的构建import numpy as npimport randomimport timeimport tqdmimport dglimport sysimport osnum_walks_per_node = 1000walk_length = 100path = './net_dbis'def construct_graph(): paper_ids = [] paper_names = [] author_ids = [] author_na原创 2022-01-26 21:18:11 · 3923 阅读 · 5 评论 -
SDNE图神经网络之小黑尝试
1.参数设定import numpy as npimport torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.utils.data import DataLoader,Datasetimport torch.optim as optimfrom argparse import ArgumentParser,ArgumentDefaultsHelpFormatterimport networkx as nx原创 2022-01-19 16:03:24 · 551 阅读 · 1 评论 -
小黑demo:TransR
import torchimport torch.nn as nnimport torch.nn.functional as Fclass TransR(nn.Module): def __init__(self,ent_tot,rel_tot,dim_e = 100,dim_r = 100,p_norm = 1,norm_flag = True,rand_init = False,margin = None): super(TransR,self).__init__()..原创 2022-02-05 11:32:45 · 1844 阅读 · 2 评论 -
The Alias sample Method(复杂度为O(1)的采样算法)
原理请见:[https://www.keithschwarz.com/darts-dice-coins/]代码运作距举例:1.初始化2.标准化(对应probs*len(probs))3.将结点1切分操作到结点3(结点1,3进行pop,进行概率切分),之后结点1放入smaller4.将结点0切分操作到结点2(结点0,2进行pop,进行概率切分),之后结点0放入larger5.将结点0切分操作到结点1(结点0,1进行pop,进行概率切分),最终完成切分任务。代码实现:def alias原创 2022-01-04 19:36:42 · 401 阅读 · 0 评论 -
小黑笔记:transe模型
1.数据集准备import openkefrom openke.data import TrainDataLoader,TestDataLoadertrain_dataloader = TrainDataLoader( in_path = "./benchmarks/FB15K237_tiny/", nbatches = 100, threads = 8, # 负采样 sampling_mode = 'normal', # bern构建负样本方式原创 2022-02-04 11:33:24 · 2883 阅读 · 2 评论 -
Line(图神经网络)之小黑尝试
1.参数初始化import argparsefrom utils.line import Linefrom tqdm import trangeimport torchimport sysimport pickleimport numpy as npfrom decimal import Decimal# 使用parser加载信息parser = argparse.ArgumentParser()# 输入文件parser.add_argument('-g','--graph_path原创 2022-01-14 16:16:29 · 1888 阅读 · 0 评论 -
小黑回来继续重振旗鼓:GAT我回来了
2.数据处理邻接矩阵与特征标准化加载数据并处理3.模型定义GraphAttentionLayerGAT整体模型多头注意力合并:4.模型训练输出:Epoch:0001 loss_train:1.9473 acc_train:0.1786 loss_val:1.8979 acc_val:0.4433 time:10.4272sEpoch:0002 loss_train:1.8933 acc_train:0.2857 loss_val:1.8609 acc_val:0.5000 ti原创 2022-06-14 08:31:37 · 229 阅读 · 0 评论