union commit

.gitignore

@@ -1,2 +1,3 @@
+# .gitignore
 reference/
 .venv/

.python-version

@@ -1 +1,2 @@
+.python-version
 3.12

README.md

@@ -1,3 +1,4 @@
+<!-- README.md -->
 <div align="center">
 
 <p align="center">

README_zh.md

@@ -1,3 +1,4 @@
+<!-- README_zh.md -->
 <div align="center">
 
 <p align="center">

TODO.md

@@ -1,3 +1,4 @@
+<!-- TODO.md -->
 # TODO — Geo-Layout-Transformer 🚀
 
 目的：遍历项目并把发现的未实现/待完善项整理到此文件，方便后续开发分配与跟踪。📝

configs/default.yaml

@@ -1,3 +1,4 @@
+# configs/default.yaml
 # Default Configuration for Geo-Layout Transformer
 
 # 1. Data Preprocessing

configs/hotspot_detection.yaml

@@ -1,3 +1,4 @@
+# configs/hotspot_detection.yaml
 # Hotspot Detection Task Configuration
 
 # Inherits from default.yaml

main.py

@@ -1,3 +1,4 @@
+# main.py
 import argparse
 from torch.utils.data import random_split
 

requirements.txt

@@ -1,3 +1,4 @@
+# requirements.txt
 torch
 torch-geometric
 gdstk

scripts/preprocess_gds.py

@@ -1,3 +1,4 @@
+# scripts/preprocess_gds.py
 import argparse
 import os
 from tqdm import tqdm

scripts/visualize_attention.py

@@ -1,3 +1,4 @@
+# scripts/visualize_attention.py
 import argparse
 import torch
 import matplotlib.pyplot as plt

src/data/dataset.py

@@ -1,3 +1,4 @@
+# src/data/dataset.py
 import torch
 from torch_geometric.data import Dataset, InMemoryDataset
 import os

src/data/gds_parser.py

@@ -1,3 +1,4 @@
+# src/data/gds_parser.py
 from typing import List, Dict, Tuple
 import gdstk
 import numpy as np

src/data/graph_constructor.py

@@ -1,3 +1,4 @@
+# src/data/graph_constructor.py
 from typing import List, Dict, Tuple
 import torch
 from torch_geometric.data import Data

src/data/init.py

@@ -0,0 +1 @@
+# src/data/init.py

src/engine/evaluator.py

@@ -1,3 +1,4 @@
+# src/engine/evaluator.py
 import torch
 from torch_geometric.data import DataLoader
 from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, roc_auc_score

src/engine/init.py

@@ -0,0 +1 @@
+# src/engine/init.py

src/engine/self_supervised.py

@@ -1,3 +1,4 @@
+# src/engine/self_supervised.py
 import torch
 import torch.nn as nn
 from torch.optim import AdamW

src/engine/trainer.py

@@ -1,3 +1,4 @@
+# src/engine/trainer.py
 import torch
 import torch.nn as nn
 from torch.optim import Adam, AdamW

src/init.py

@@ -0,0 +1 @@
+# src/init.py

src/models/geo_layout_transformer.py

@@ -1,3 +1,4 @@
+# src/models/geo_layout_transformer.py
 import torch
 import torch.nn as nn
 from .gnn_encoder import GNNEncoder

src/models/gnn_encoder.py

@@ -1,3 +1,4 @@
+# src/models/gnn_encoder.py
 import torch
 import torch.nn as nn
 from torch_geometric.nn import GCNConv, SAGEConv, GATConv, global_mean_pool

src/models/init.py

@@ -0,0 +1 @@
+# src/models/init.py

src/models/task_heads.py

@@ -1,3 +1,4 @@
+# src/models/task_heads.py
 import torch
 import torch.nn as nn
 

src/models/transformer_core.py

@@ -1,3 +1,4 @@
+# src/models/transformer_core.py
 import torch
 import torch.nn as nn
 import math

src/utils/config_loader.py

@@ -1,3 +1,4 @@
+# src/utils/config_loader.py
 import yaml
 from pathlib import Path
 

src/utils/init.py

@@ -0,0 +1 @@
+# src/utils/init.py

src/utils/logging.py

@@ -1,3 +1,4 @@
+# src/utils/logging.py
 import logging
 import sys
 

uv.lock

@@ -1,3 +1,4 @@
+# uv.lock
 version = 1
 revision = 2
 requires-python = ">=3.12"