popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

integrating new CNN model

Browse Source
This commit is contained in:
Dobromir Popov
2025-07-23 16:59:35 +03:00
parent 1be270cc5c
commit f1d63f9da6
15 changed files with 1896 additions and 1003 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
NN/training/integrate_checkpoint_management.py
View File

@ -40,7 +40,7 @@ from utils.training_integration import get_training_integration
# Import training components
from NN.models.dqn_agent import DQNAgent
from NN.models.cnn_model import CNNModelTrainer, create_enhanced_cnn_model
from NN.models.standardized_cnn import StandardizedCNN
from core.extrema_trainer import ExtremaTrainer
from core.negative_case_trainer import NegativeCaseTrainer
from core.data_provider import DataProvider
@ -100,18 +100,10 @@ class CheckpointIntegratedTrainingSystem:
)
logger.info("✅ DQN Agent initialized with checkpoint management")
# Initialize CNN Model with checkpoint management
logger.info("Initializing CNN Model with checkpoints...")
cnn_model, self.cnn_trainer = create_enhanced_cnn_model(
input_size=60,
feature_dim=50,
output_size=3
)
# Update trainer with checkpoint management
self.cnn_trainer.model_name = "integrated_cnn_model"
self.cnn_trainer.enable_checkpoints = True
self.cnn_trainer.training_integration = self.training_integration
logger.info("✅ CNN Model initialized with checkpoint management")
# Initialize StandardizedCNN Model with checkpoint management
logger.info("Initializing StandardizedCNN Model with checkpoints...")
self.cnn_model = StandardizedCNN(model_name="integrated_cnn_model")
logger.info("✅ StandardizedCNN Model initialized with checkpoint management")
# Initialize ExtremaTrainer with checkpoint management
logger.info("Initializing ExtremaTrainer with checkpoints...")