深度学习
与非学堂
这个作者很懒,什么都没留下…
展开
-
William Hamilton对GraphSAGE的代码实现
以下是大牛Hamilton对GraphSAGE的代码实现,仅作备忘。 首先是文件结构: ├── aggregators.py ├── cora │ ├── cora.cites │ ├── cora.content │ └── README ├── encoders.py ├── __init__.py ├── model.py ├── pubmed-data │ ├── Pubmed-Diabetes.DIRECTED.cites.tab │ ├── Pubmed-Diabete...原创 2021-10-19 06:34:18 · 585 阅读 · 0 评论 -
PyG从networkx导入数据的节点名称问题
最近在研究使用PyG进行图数据处理时,发现PyG从networkx导入数据时,原有节点的名字都被转为了整数,而后面还要把训练的节点嵌入与原有节点名字标签一一对应起来。 经过一番探索,发现是PyG的from_networkx函数在实现时,用到了networkx.relabel.convert_node_labels_to_integers函数: def from_networkx(G, group_node_attrs: Optional[Union[List[str], all]] = None,原创 2021-10-18 20:52:29 · 1693 阅读 · 0 评论 -
PyG的第一个示例:下载数据后使用GCN训练
PyG官网的例子,先让代码跑起来! 注意在 import torch import torch.nn.functional as F from torch_geometric.nn import GCNConv from torch_geometric.datasets import Planetoid #下载Cora数据集 dataset = Planetoid(root="../dataset/Cora", name="Cora") print(len(dataset)) print(dat原创 2021-09-29 23:41:12 · 598 阅读 · 0 评论 -
使用RNN模型构建字符串批量转换功能seq2seq
使用RNN Module构建的一个字符串转换功能: import torch import torch.optim as optim class Model(torch.nn.Module): """ RNN """ def __init__(self, input_size, hidden_size, batch_size,num_layers): super(Model, self).__init__() self.batch_siz原创 2021-09-28 23:24:38 · 131 阅读 · 0 评论 -
RNN序列到序列(seq2seq)生成代码
使用RNNCell构建模型,创建一个初级的RNN序列到序列(seq2seq)模型的生成代码。 import torch import torch.optim as optim class Model(torch.nn.Module): """ RNNCell """ def __init__(self, input_size, hidden_size, batch_size): super(Model, self).__init__()原创 2021-09-28 19:48:05 · 511 阅读 · 0 评论