20 lines
824 B
Python
20 lines
824 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
|
|
- 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
|
|
|
|
__all__ = ['CNNModel', 'DQNAgent', 'MassiveRLNetwork', 'COBRLModelInterface', 'AdvancedTradingTransformer', 'TradingTransformerConfig']
|