45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script to verify COB RL model extraction
|
|
"""
|
|
|
|
print('=== COB RL MODEL AUDIT ===')
|
|
|
|
# Test the extracted COB RL model
|
|
try:
|
|
from NN.models.cob_rl_model import MassiveRLNetwork, COBRLModelInterface
|
|
|
|
print('1. Testing MassiveRLNetwork...')
|
|
model = MassiveRLNetwork()
|
|
model_info = model.get_model_info()
|
|
|
|
print(' ✓ Model created successfully')
|
|
print(f' ✓ Total parameters: {model_info["total_parameters"]:,}')
|
|
print(f' ✓ Architecture: {model_info["architecture"]}')
|
|
|
|
print('2. Testing COBRLModelInterface...')
|
|
interface = COBRLModelInterface()
|
|
print(f' ✓ Interface created on device: {interface.device}')
|
|
|
|
print('3. Testing imports from models module...')
|
|
from NN.models import MassiveRLNetwork as ImportedModel
|
|
print(' ✓ Models properly exported')
|
|
|
|
print()
|
|
print('=== AUDIT SUMMARY ===')
|
|
print('✅ COB RL MODEL EXTRACTION SUCCESSFUL')
|
|
print('🧠 The 1B parameter model is now in: NN/models/cob_rl_model.py')
|
|
print(' - MassiveRLNetwork: Core transformer model')
|
|
print(' - COBRLModelInterface: Training/inference interface')
|
|
print()
|
|
print('📊 Model Details:')
|
|
print(f' - Parameters: {model_info["total_parameters"]:,} (~1B)')
|
|
print(f' - Input size: {model_info["input_size"]} (COB features)')
|
|
print(f' - Hidden size: {model_info["hidden_size"]}')
|
|
print(f' - Layers: {model_info["num_layers"]}')
|
|
print(' - Output: Price direction, value, confidence')
|
|
|
|
except Exception as e:
|
|
print(f'❌ Audit failed: {e}')
|
|
import traceback
|
|
traceback.print_exc() |