图神经网络GNN

pou6

已于 2024-04-24 16:24:14 修改

阅读量256

点赞数 2

文章标签：神经网络 python pytorch

于 2024-04-21 21:19:01 首次发布

本文链接：https://blog.csdn.net/m0_66184663/article/details/138045611

版权

本文介绍了如何使用PyTorch和相关库如networkx和torch_geometric对KarateClub社交网络数据集进行分析，包括构建图模型、GCN层实现、以及节点嵌入的可视化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
import networkx as nx
import matplotlib.pyplot as plt
from torch_geometric.datasets import KarateClub
from torch_geometric.utils import to_networkx
from torch.nn import Linear
from torch_geometric.nn import GCNConv


def visualize_graph(G,color):
    plt.figure(figsize=(7,7))
    plt.xticks([])
    plt.yticks([])
    nx.draw_networkx(G,pos=nx.spring_layout(G,seed=42),with_labels=False,node_color=color,cmap='Set2')

    plt.show()

def visualize_embedding(h,color,epoch=None,loss=None):
    plt.figure(figsize=(7,7))
    plt.xticks([])
    plt.yticks([])
    h=h.detach().cpu().numpy()
    plt.scatter(h[:,0],h[:,1],s=140,c=color,cmap='Set2')
    if epoch is not None and loss is not None:
        plt.xlabel(f'Epoch:{epoch},Loss:{loss.item():.4f}',fontsize=16)
    plt.show()



class GCN(torch.nn.Module):
    def __init__(self):
        super().__init__()
        to

最低0.47元/天解锁文章