Upload GraphSemCom GNN encoder + edge relation classifier

README.md

@@ -0,0 +1,24 @@
+---
+license: mit
+tags:
+- graph-neural-network
+- semantic-communication
+- webnlg
+- dart
+---
+
+# GraphSemCom GNN (rel4593)
+
+This repo contains:
+- `encoder_best.pt`: GNN encoder weights
+- `edge_clf_best.pt`: edge relation classifier weights
+- `rel2id.json`: mapping of relation string -> class id
+- `config.json`: metadata
+
+## How to use (concept)
+1. Build a graph Data(x, edge_index, edge_type)
+2. Run encoder -> h
+3. Add AWGN to h (optional)
+4. Predict relations with edge_clf(h[src], h[dst])
+
+Trained with AWGN augmentation (random SNR 0–20 dB).

config.json

@@ -0,0 +1,11 @@
+{
+  "model_type": "graphsemcom-gnn",
+  "embedding_dim": 384,
+  "num_relations": 4593,
+  "files": [
+    "encoder_best.pt",
+    "edge_clf_best.pt",
+    "rel2id.json"
+  ],
+  "notes": "GNN encoder + relation classifier trained on GEM/web_nlg(en) + Yale-LILY/dart. AWGN augmentation during training."
+}

edge_clf_best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:52d2ebfcf4e419c65439d5c574d6195aeb9d03a1dccd40d30fc71873ab005a25
+size 12575257

encoder_best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a9fd18189e31b583904372a0670c1526b83ef03562d823cea6f79232242ddfa
+size 17663709