28 lines
1.2 KiB
Python
28 lines
1.2 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.
|
|
"""
|
|
|
|
# Import core models
|
|
from NN.models.dqn_agent import DQNAgent, MassiveRLNetwork
|
|
from NN.models.cob_rl_model import COBRLModelInterface
|
|
from NN.models.advanced_transformer_trading import AdvancedTradingTransformer, TradingTransformerConfig
|
|
from NN.models.standardized_cnn import StandardizedCNN # Use the unified CNN model
|
|
|
|
# Import model interfaces
|
|
from NN.models.model_interfaces import ModelInterface, CNNModelInterface, RLAgentInterface, ExtremaTrainerInterface
|
|
|
|
# Export the unified StandardizedCNN as CNNModel for compatibility
|
|
CNNModel = StandardizedCNN
|
|
|
|
__all__ = ['CNNModel', 'StandardizedCNN', 'DQNAgent', 'MassiveRLNetwork', 'COBRLModelInterface', 'AdvancedTradingTransformer', 'TradingTransformerConfig',
|
|
'ModelInterface', 'CNNModelInterface', 'RLAgentInterface', 'ExtremaTrainerInterface']
|