|
5 | 5 | "id": "7a56d467", |
6 | 6 | "metadata": {}, |
7 | 7 | "source": [ |
8 | | - "# 🧠 GNN for Node-Level Regression: Predicting Sine Waves\n", |
| 8 | + "# 🌐 GNN for Node-Level Regression: Predicting Sine Waves\n", |
9 | 9 | "\n", |
10 | 10 | "**Author:** Loïc Bachelot \n", |
11 | 11 | "**Goal:** This notebook demonstrates how to use a GNN to predict 1D sine wave signals at each node of a synthetic graph.\n", |
|
491 | 491 | "- **`signal`**: Node signals (sine waves delayed by distance to a hidden origin) `[num_nodes, signal_length]`\n", |
492 | 492 | "- **`origin`**: The hidden 2D origin point `[2]` used to generate node signals (not directly used for supervision here)\n", |
493 | 493 | "- **`edge_index`**: Connectivity information between nodes (edges)\n", |
494 | | - "- **`edge_weight`**: Inverse-distance weights for edges (higher weight = closer nodes)\n", |
| 494 | + "- **`edge_attr`**: distance weights for edges (both in x and y)\n", |
495 | 495 | "\n", |
496 | 496 | "This compact graph-based representation allows flexible training for node-level prediction tasks, \n", |
497 | 497 | "while preserving spatial and relational structure essential for learning.\n" |
|
632 | 632 | "### 🏗️ Architecture Overview:\n", |
633 | 633 | "\n", |
634 | 634 | "1. **Signal Feature Extraction** \n", |
635 | | - " Each node’s raw signal is processed through a 1D convolution and max pooling to extract compact local features.\n", |
| 635 | + " Each node’s raw signal is processed through a simple MLP to extract compact local features.\n", |
636 | 636 | "\n", |
637 | 637 | "2. **Feature Fusion** \n", |
638 | 638 | " The extracted signal features are concatenated with the node's spatial coordinates (x, y) to create a combined feature vector.\n", |
|
0 commit comments