![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
小黑之trans系列
爱喝喜茶爱吃烤冷面的小黑黑
这个作者很懒,什么都没留下…
展开
-
小黑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 评论 -
小黑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 评论 -
小黑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 评论 -
小黑笔记: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 评论