Hugging face(4) Bert模型及Collator与Trainer
1. Collator数据处理 目的：将dataset的初始数据进行规范化批量处理，用以后续的前向计算 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22...
深度学习
Faiss向量数据库相似搜索
Faiss (Facebook AI Similarity Search) 是一个用于高效相似向量搜索的库，特别适合处理大规模向量数据。 它的核心功能是给定一个向量数据库，为当前的query vector(s) 寻找Top-K个...
图神经网络DGL-01-DGL基础
homogeneous graph ： 一种节点，一种边 heterogeneous graph: 多种节点，多种边 bipartite graph : 两种节点，一种边 1、DGLgraph结构 1.1 同构图 Cora论文关系图–同构图(ho...
图神经网络DGL-02同构图_节点分类
1 2 3 4 5 6 7 8 import dgl import dgl.nn as dglnn import torch import torch.nn as nn import torch.nn.functional as F import numpy as np 0、预测任务与数据 预测论文属于哪一种类别，即为多分类问题 1 2 3 4 5 6 7 8 9 10 11 12 13 14...
图神经网络DGL-03同构图_边回归
（1）边回归问题，对图中已存在的边的定量(回归)/定性(分类)，结合训练得到的边的embedding做GNN神经网络预测。 （2）边的embe...
图神经网络DGL-03同构图_边预测
边的预测问题即预测两个节点间是否可能存在边，可以视为二分类问题。 在训练时，首先同样要先更新节点信息，然后计算边的特征— 将图中已知...