diff --git a/.gitignore b/.gitignore index be0cf26..be9d7c4 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ +# .gitignore reference/ .venv/ diff --git a/.python-version b/.python-version index e4fba21..57c7ce8 100644 --- a/.python-version +++ b/.python-version @@ -1 +1,2 @@ +.python-version 3.12 diff --git a/README.md b/README.md index c514ab8..971e6f7 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,4 @@ +

diff --git a/README_zh.md b/README_zh.md index 42cc878..a961788 100644 --- a/README_zh.md +++ b/README_zh.md @@ -1,3 +1,4 @@ +

diff --git a/TODO.md b/TODO.md index 2b55911..4fc7a0b 100644 --- a/TODO.md +++ b/TODO.md @@ -1,3 +1,4 @@ + # TODO — Geo-Layout-Transformer 🚀 目的：遍历项目并把发现的未实现/待完善项整理到此文件，方便后续开发分配与跟踪。📝 diff --git a/configs/default.yaml b/configs/default.yaml index 1e22f53..ea6e42a 100644 --- a/configs/default.yaml +++ b/configs/default.yaml @@ -1,3 +1,4 @@ +# configs/default.yaml # Default Configuration for Geo-Layout Transformer # 1. Data Preprocessing diff --git a/configs/hotspot_detection.yaml b/configs/hotspot_detection.yaml index d1cf0a9..20b1f37 100644 --- a/configs/hotspot_detection.yaml +++ b/configs/hotspot_detection.yaml @@ -1,3 +1,4 @@ +# configs/hotspot_detection.yaml # Hotspot Detection Task Configuration # Inherits from default.yaml diff --git a/main.py b/main.py index 6d670ab..6086b0b 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,4 @@ +# main.py import argparse from torch.utils.data import random_split diff --git a/requirements.txt b/requirements.txt index e85ec6b..6c3005d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ +# requirements.txt torch torch-geometric gdstk diff --git a/scripts/preprocess_gds.py b/scripts/preprocess_gds.py index 743ef9a..68803b2 100644 --- a/scripts/preprocess_gds.py +++ b/scripts/preprocess_gds.py @@ -1,3 +1,4 @@ +# scripts/preprocess_gds.py import argparse import os from tqdm import tqdm diff --git a/scripts/visualize_attention.py b/scripts/visualize_attention.py index 7e85aab..635d0b3 100644 --- a/scripts/visualize_attention.py +++ b/scripts/visualize_attention.py @@ -1,3 +1,4 @@ +# scripts/visualize_attention.py import argparse import torch import matplotlib.pyplot as plt diff --git a/src/data/dataset.py b/src/data/dataset.py index 9f8c9d9..02affd8 100644 --- a/src/data/dataset.py +++ b/src/data/dataset.py @@ -1,3 +1,4 @@ +# src/data/dataset.py import torch from torch_geometric.data import Dataset, InMemoryDataset import os diff --git a/src/data/gds_parser.py b/src/data/gds_parser.py index a61993b..36ae17a 100644 --- a/src/data/gds_parser.py +++ b/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 diff --git a/src/data/graph_constructor.py b/src/data/graph_constructor.py index 9a2efe5..b88ef50 100644 --- a/src/data/graph_constructor.py +++ b/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 diff --git a/src/data/init.py b/src/data/init.py index e69de29..812cd58 100644 --- a/src/data/init.py +++ b/src/data/init.py @@ -0,0 +1 @@ +# src/data/init.py diff --git a/src/engine/evaluator.py b/src/engine/evaluator.py index ce71853..d561a39 100644 --- a/src/engine/evaluator.py +++ b/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 diff --git a/src/engine/init.py b/src/engine/init.py index e69de29..acc8ceb 100644 --- a/src/engine/init.py +++ b/src/engine/init.py @@ -0,0 +1 @@ +# src/engine/init.py diff --git a/src/engine/self_supervised.py b/src/engine/self_supervised.py index eae53af..8a3b508 100644 --- a/src/engine/self_supervised.py +++ b/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 diff --git a/src/engine/trainer.py b/src/engine/trainer.py index ec0ea5f..0a26176 100644 --- a/src/engine/trainer.py +++ b/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 diff --git a/src/init.py b/src/init.py index e69de29..bf41fd2 100644 --- a/src/init.py +++ b/src/init.py @@ -0,0 +1 @@ +# src/init.py diff --git a/src/models/geo_layout_transformer.py b/src/models/geo_layout_transformer.py index 0c94032..34a9a88 100644 --- a/src/models/geo_layout_transformer.py +++ b/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 diff --git a/src/models/gnn_encoder.py b/src/models/gnn_encoder.py index 0c9c285..1140c3f 100644 --- a/src/models/gnn_encoder.py +++ b/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 diff --git a/src/models/init.py b/src/models/init.py index e69de29..5062d42 100644 --- a/src/models/init.py +++ b/src/models/init.py @@ -0,0 +1 @@ +# src/models/init.py diff --git a/src/models/task_heads.py b/src/models/task_heads.py index 74c311f..fd374eb 100644 --- a/src/models/task_heads.py +++ b/src/models/task_heads.py @@ -1,3 +1,4 @@ +# src/models/task_heads.py import torch import torch.nn as nn diff --git a/src/models/transformer_core.py b/src/models/transformer_core.py index b96f34d..1154223 100644 --- a/src/models/transformer_core.py +++ b/src/models/transformer_core.py @@ -1,3 +1,4 @@ +# src/models/transformer_core.py import torch import torch.nn as nn import math diff --git a/src/utils/config_loader.py b/src/utils/config_loader.py index eaa8abc..4171c0e 100644 --- a/src/utils/config_loader.py +++ b/src/utils/config_loader.py @@ -1,3 +1,4 @@ +# src/utils/config_loader.py import yaml from pathlib import Path diff --git a/src/utils/init.py b/src/utils/init.py index e69de29..78539f9 100644 --- a/src/utils/init.py +++ b/src/utils/init.py @@ -0,0 +1 @@ +# src/utils/init.py diff --git a/src/utils/logging.py b/src/utils/logging.py index be51c24..45643e7 100644 --- a/src/utils/logging.py +++ b/src/utils/logging.py @@ -1,3 +1,4 @@ +# src/utils/logging.py import logging import sys diff --git a/uv.lock b/uv.lock index 981fb63..d11ebc7 100644 --- a/uv.lock +++ b/uv.lock @@ -1,3 +1,4 @@ +# uv.lock version = 1 revision = 2 requires-python = ">=3.12"