gogo2/NN/models/__init__.py
2025-03-25 12:48:58 +02:00

14 lines
583 B
Python

"""
Neural Network Models
====================
This package contains the neural network models used in the trading system:
- CNN Model: Deep convolutional neural network for feature extraction
- Transformer Model: Processes high-level features for improved pattern recognition
- MoE: Mixture of Experts model that combines multiple neural networks
"""
from NN.models.cnn_model import CNNModel
from NN.models.transformer_model import TransformerModel, TransformerBlock, MixtureOfExpertsModel
__all__ = ['CNNModel', 'TransformerModel', 'TransformerBlock', 'MixtureOfExpertsModel']