20 lines
380 B
Python
20 lines
380 B
Python
|
"""
|
||
|
|
Surrogate models for FEA acceleration.
|
||
|
|
|
||
|
|
Available surrogates:
|
||
|
|
- EnsembleSurrogate: Multiple MLPs with uncertainty quantification
|
||
|
|
- OODDetector: Out-of-distribution detection
|
||
|
|
"""
|
||
|
|
|
||
|
|
from .ensemble_surrogate import (
|
||
|
|
EnsembleSurrogate,
|
||
|
|
OODDetector,
|
||
|
|
create_and_train_ensemble
|
||
|
|
)
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'EnsembleSurrogate',
|
||
|
|
'OODDetector',
|
||
|
|
'create_and_train_ensemble'
|
||
|
|
]
|