cleanup, CNN fixes

NN/models/__init__.py

@@ -4,17 +4,16 @@ 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
+- 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_pytorch import EnhancedCNNModel as CNNModel
-from NN.models.transformer_model_pytorch import (
-    TransformerModelPyTorch as TransformerModel,
-    MixtureOfExpertsModelPyTorch as MixtureOfExpertsModel
-)
+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', 'TransformerModel', 'MixtureOfExpertsModel', 'MassiveRLNetwork', 'COBRLModelInterface']
+__all__ = ['CNNModel', 'DQNAgent', 'MassiveRLNetwork', 'COBRLModelInterface', 'AdvancedTradingTransformer', 'TradingTransformerConfig']