1. Job Roadmap
  2. Home
  3. Problem Set
  4. codenotelist
  5. Forum
  6. course
  7. Shore Share Sessions
  8. Record
  1. Login
  2. Sign Up
  3. Language
    1. English
    2. 한국어
    3. 简体中文
    4. 正體中文
    ZhContent TextSol AI分析

好的,这是一个关于实现单层 GraphSAGE-Mean 节点分类器的题解和 Python 代码。

思路分析

该题目要求我们从零开始,仅使用 numpy、pandas 或 scikit-learn 库,实现一个完整的图神经网络节点分类流程。整个流程可以分解为以下几个核心步骤:

  1. 数据加载与图表示:

    • 从标准输入读取单行 JSON 字符串,并解析成我们需要的数据结构。

P3684.第2题-简易GNN的实现

    1000ms Tried: 33 Accepted: 8 Difficulty: 5 所属公司 : 蚂蚁
    算法与标签>机器学习算法

题目内容

请在仅使用 numpy/pandas/scikit−learnnumpy/pandas/scikit-learnnumpy/pandas/scikit−learn 的前提下,实现一个单层 GraphSAGE−MeanGraphSAGE-MeanGraphSAGE−Mean 节点分类器。

流程如下:

1.输入图:无向、无自环,给定邻接表 edgesedgesedges ; 节点按索引 0...N−10...N-10...N−1。

2.原始特征 X∈RN×dX \in \mathbb{R}^{N \times d}X∈RN×d 直接读取。

3.一跳平均聚合:hi=1∣N(i)∣+1(xi+∑j∈N(i)xj)h_{i}=\frac{1}{|\mathcal{N}(i)|+1}\left(x_{i}+\sum_{j \in \mathcal{N}(i)} x_{j}\right)hi​=∣N(i)∣+11​(xi​+∑j∈N(i)​xj​) 得到 H∈RN×dH \in \mathbb{R}^{N \times d}H∈RN×d 。若某节点度数为 000 ,则 hi=xih_i= x_ihi​=xi​ 。

4.线性分类 +Sigmoid+Sigmoid+Sigmoid

令 y∈y∈y∈{0,10,10,1} 仅在 traintraintrain_ idxidxidx 上已知,目标 min⁡w12∥Hw−y∥22+λ∥w∥22,λ=0.01\min _{w} \frac{1}{2}\|H w-y\|_{2}^{2}+\lambda\|w\|_{2}^{2},\lambda=0.01minw​21​∥Hw−y∥22​+λ∥w∥22​,λ=0.01 。

闭式解 w=(HTTHT+λI)−1HTTyTw=\left(H_{T}^T H_{T}+\lambda I\right)^{-1}H^T_TyTw=(HTT​HT​+λI)−1HTT​yT ,其中下标 TTT 表示只取训练索引行。

预测概率 pi=σ(hiTw)p_i=σ(h^T_iw)pi​=σ(hiT​w) ,阈值 0.50.50.5 得标签。

输入描述

单行 JSONJSONJSON :

{
 "nodes":[[f11,f12,...],// N×d 原始特征
     ...]
 "edges":[[u1,v1],[u2,v2],...],// 无向边,端点索引
 "train_idx":[i1,i2,...],   //已知标签节点下标
 "train_y":[0/1,...],       //与 train_idx 对应
 "test_idx":[j1,j2,...]      //需预测节点下标
}

约束:

  • N≤20,d≤5,[edges]≤30,[trainN≤20,d≤5,[edges]≤30,[trainN≤20,d≤5,[edges]≤30,[train_idx]≥4,[testidx]≥4,[testidx]≥4,[test _idx]≤8idx]≤8idx]≤8

  • traintraintrain_idx∩testidx∩testidx∩test _idx=∅idx=∅idx=∅;两类比例 252525%-757575%

输出描述

仅一行 JSONJSONJSON :

{
 "weights":[w1,w2,...],//长度 d,6位小数,使用 round(x,6)即可
 "test_proba":[p1,...],  //长度=[test_idx],6位小数,使用 round(x,6)即可
 "test_pred":[0/1,...]
}

补充说明

1.邻接矩阵稀疏,但 N、dN、dN、d 很小,直接用 NumPyNumPyNumPy 逐行遍历即可。

2.理论上 edgesedgesedges 无重复;若输入出现重复也应正常处理(去重后计算)。

3.为确保通过测试用例,仅允许使用 numpy/pandas/scikit−leannumpy/pandas/scikit-leannumpy/pandas/scikit−lean 实现。

样例1

输入

{"nodes":[[0,0],[0.2,0.1],[0.1,-0,1],[4,4],[4.2,3.9]],"edges":[[0.1],[0,2],[3,4],"train_idx":[0,3],"train_y":[0,1],"test_idx":[1,4]}

输出

{"weights":[0.085354,0.163463],"test_pioba":[0.50419,0.730977],"test_pred":[1,1]}

登录后即可使用 AI 分析。

模式
倒计时时长
:

最长 10 小时 59 分;应用后按此时长重新开始。

提示:点击提交记录在左侧题面区域查看详情
题库
AI分析设置
留空使用官方API Key,每天有次数限制(自定义API Key仅限会员和管理员使用,不限次数)
会员和管理员可切换模型;切到 Kimi/智谱/通义/豆包时需填写对应供应商 API Key
升级会员,可将运行与提交冷却时间缩短至 1 秒起

Status

  • Judging Queue
  • Service Status

Development

  • Open Source

Support

  • Help
  • Contact Us

About

  • About
  • Privacy
  • Terms of Service
  • Copyright Complaint
  1. Language
    1. English
    2. 한국어
    3. 简体中文
    4. 正體中文
  2. Legacy mode
  3. Theme
    1. Light
    2. Dark
  1. 京ICP备2025123107号-1
  2. Worker 0, 55ms
  3. Powered by Hydro v5.0.0-beta.18 Community
CLOSE


ScanQRCodePrompt

请使用微信扫描下方二维码完成注册

Forgot password or username?