Skip to content

Commit 7805841

Browse files
authored
Merge pull request #28 from loicbachelot/main
[update] finish cleanup for geolab
2 parents 3ffd813 + 01c0ad2 commit 7805841

10 files changed

+115
-1868
lines changed

_toc.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ parts:
1111
- caption: Seismic Phase Detection
1212
chapters:
1313
- file: notebooks/niyiyu/curated_pnw_dataset_seisbench.ipynb
14-
- file: notebooks/Loic/UNet/add_training_split.ipynb
1514
- file: notebooks/Loic/UNet/make_unet_annotated.ipynb
1615
- file: notebooks/Amanda/unet_performance_metrics.ipynb
1716
- file: notebooks/Amanda/apply_unet.ipynb

notebooks/Loic/GNN/GNN_node_regression_annotated.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
"id": "7a56d467",
66
"metadata": {},
77
"source": [
8-
"# 🧠 GNN for Node-Level Regression: Predicting Sine Waves\n",
8+
"# 🌐 GNN for Node-Level Regression: Predicting Sine Waves\n",
99
"\n",
1010
"**Author:** Loïc Bachelot \n",
1111
"**Goal:** This notebook demonstrates how to use a GNN to predict 1D sine wave signals at each node of a synthetic graph.\n",
@@ -491,7 +491,7 @@
491491
"- **`signal`**: Node signals (sine waves delayed by distance to a hidden origin) `[num_nodes, signal_length]`\n",
492492
"- **`origin`**: The hidden 2D origin point `[2]` used to generate node signals (not directly used for supervision here)\n",
493493
"- **`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",
495495
"\n",
496496
"This compact graph-based representation allows flexible training for node-level prediction tasks, \n",
497497
"while preserving spatial and relational structure essential for learning.\n"
@@ -632,7 +632,7 @@
632632
"### 🏗️ Architecture Overview:\n",
633633
"\n",
634634
"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",
636636
"\n",
637637
"2. **Feature Fusion** \n",
638638
" The extracted signal features are concatenated with the node's spatial coordinates (x, y) to create a combined feature vector.\n",

notebooks/Loic/UNet/add_training_split.ipynb

Lines changed: 0 additions & 72 deletions
This file was deleted.

0 commit comments

Comments
 (0)