深度学习
与非学堂
这个作者很懒,什么都没留下…
展开
-
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 · 562 阅读 · 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 · 1646 阅读 · 0 评论 -
PyG的第一个示例:下载数据后使用GCN训练
PyG官网的例子,先让代码跑起来!注意在import torchimport torch.nn.functional as Ffrom torch_geometric.nn import GCNConvfrom torch_geometric.datasets import Planetoid#下载Cora数据集dataset = Planetoid(root="../dataset/Cora", name="Cora")print(len(dataset))print(dat原创 2021-09-29 23:41:12 · 584 阅读 · 0 评论 -
使用RNN模型构建字符串批量转换功能seq2seq
使用RNN Module构建的一个字符串转换功能:import torchimport torch.optim as optimclass 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 · 126 阅读 · 0 评论 -
RNN序列到序列(seq2seq)生成代码
使用RNNCell构建模型,创建一个初级的RNN序列到序列(seq2seq)模型的生成代码。import torchimport torch.optim as optimclass Model(torch.nn.Module): """ RNNCell """ def __init__(self, input_size, hidden_size, batch_size): super(Model, self).__init__()原创 2021-09-28 19:48:05 · 500 阅读 · 0 评论