Atomizer/atomizer-field/neural_models/__init__.py

"""
AtomizerField Neural Models Package

Phase 2: Neural Network Architecture for Field Prediction

This package contains neural network models for learning complete FEA field results
from mesh geometry, boundary conditions, and loads.

Models:
- AtomizerFieldModel: Full field predictor (displacement + stress fields)
- ParametricFieldPredictor: Design-conditioned scalar predictor (mass, freq, disp, stress)
"""

__version__ = "2.0.0"

# Import main model classes for convenience
from .field_predictor import AtomizerFieldModel, create_model
from .parametric_predictor import ParametricFieldPredictor, create_parametric_model

__all__ = [
    'AtomizerFieldModel',
    'create_model',
    'ParametricFieldPredictor',
    'create_parametric_model',
]