Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -26,4 +26,4 @@ data/
Thumbs.db

stft_dataset.npz
*.onnx
*.onnx
56 changes: 56 additions & 0 deletions subsystem2_1d/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
# subsystem2_1d/__init__.py
# =========================================================
# Subsystem 2 -- 1D Time-Wave Preprocessing, Models & Training
# Person 2 | feature/1d-model-p2
# =========================================================
# Public API -- consumers can import cleanly as:
# from subsystem2_1d import normalize_1d_iq, IQAugmentation
# from subsystem2_1d import get_1d_dataloaders, RFSignalDataset1D
# from subsystem2_1d import CNN1DClassifier, CNNTransformerHybrid
# from subsystem2_1d import build_model
# from subsystem2_1d import EarlyStopping, TrainingConfig, train_model
# from subsystem2_1d import evaluate_model, print_edge_suitability_report

from subsystem2_1d.normalization_1d import normalize_1d_iq, IQAugmentation
from subsystem2_1d.dataloader_1d import get_1d_dataloaders, RFSignalDataset1D
from subsystem2_1d.classifiers_1d import (
CNN1DClassifier,
CNNTransformerHybrid,
PositionalEncoding,
build_model,
)
from subsystem2_1d.train_1d import (
EarlyStopping,
TrainingConfig,
train_model,
)
from subsystem2_1d.eval_1d import (
evaluate_model,
print_edge_suitability_report,
load_model_from_checkpoint,
measure_inference_latency,
)

__all__ = [
# Preprocessing & normalization
"normalize_1d_iq",
"IQAugmentation",
# DataLoader pipeline
"get_1d_dataloaders",
"RFSignalDataset1D",
# Model architectures
"CNN1DClassifier",
"CNNTransformerHybrid",
"PositionalEncoding",
"build_model",
# Training pipeline
"EarlyStopping",
"TrainingConfig",
"train_model",
# Evaluation pipeline
"evaluate_model",
"print_edge_suitability_report",
"load_model_from_checkpoint",
"measure_inference_latency",
]

3 changes: 3 additions & 0 deletions subsystem2_1d/checkpoints/.gitkeep
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# This directory holds saved model checkpoints (.pt / .pth / .ckpt).
# All checkpoint files are gitignored — see root .gitignore.
# Only this .gitkeep file is tracked to preserve the directory in git.
Binary file added subsystem2_1d/checkpoints/best_cnn_transformer.pt
Binary file not shown.
Binary file added subsystem2_1d/checkpoints/best_model.pt
Binary file not shown.
Loading
Loading