21 lines
799 B
Python
21 lines
799 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
|
|
|
|
PyTorch implementation only.
|
|
"""
|
|
|
|
from NN.models.cnn_model_pytorch import EnhancedCNNModel as CNNModel
|
|
from NN.models.transformer_model_pytorch import (
|
|
TransformerModelPyTorch as TransformerModel,
|
|
MixtureOfExpertsModelPyTorch as MixtureOfExpertsModel
|
|
)
|
|
from NN.models.cob_rl_model import MassiveRLNetwork, COBRLModelInterface
|
|
|
|
__all__ = ['CNNModel', 'TransformerModel', 'MixtureOfExpertsModel', 'MassiveRLNetwork', 'COBRLModelInterface']
|