22 lines
1.0 KiB
Python
22 lines
1.0 KiB
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
|
|
- DQN Agent: Deep Q-Network for reinforcement learning
|
|
- COB RL Model: Specialized RL model for order book data
|
|
- Advanced Transformer: High-performance transformer for trading
|
|
|
|
PyTorch implementation only.
|
|
"""
|
|
|
|
from NN.models.cnn_model import EnhancedCNNModel as CNNModel
|
|
from NN.models.dqn_agent import DQNAgent
|
|
from NN.models.cob_rl_model import MassiveRLNetwork, COBRLModelInterface
|
|
from NN.models.advanced_transformer_trading import AdvancedTradingTransformer, TradingTransformerConfig
|
|
from NN.models.model_interfaces import ModelInterface, CNNModelInterface, RLAgentInterface, ExtremaTrainerInterface
|
|
|
|
__all__ = ['CNNModel', 'DQNAgent', 'MassiveRLNetwork', 'COBRLModelInterface', 'AdvancedTradingTransformer', 'TradingTransformerConfig',
|
|
'ModelInterface', 'CNNModelInterface', 'RLAgentInterface', 'ExtremaTrainerInterface']
|