edit test

This commit is contained in:
Dobromir Popov
2025-07-30 00:39:09 +03:00
parent c5a9e75ee7
commit 2dcb8a5e18

View File

@ -46,12 +46,13 @@ def test_dqn_architecture():
output = network(test_input) output = network(test_input)
if isinstance(output, tuple): if isinstance(output, tuple):
q_values, regime_pred, price_pred, volatility_pred, features = output q_values, regime_pred, price_pred, volatility_pred, features, multi_timeframe_pred = output
print(f" ✅ Q-values shape: {q_values.shape}") print(f" ✅ Q-values shape: {q_values.shape}")
print(f" ✅ Regime prediction shape: {regime_pred.shape}") print(f" ✅ Regime prediction shape: {regime_pred.shape}")
print(f" ✅ Price prediction shape: {price_pred.shape}") print(f" ✅ Price prediction shape: {price_pred.shape}")
print(f" ✅ Volatility prediction shape: {volatility_pred.shape}") print(f" ✅ Volatility prediction shape: {volatility_pred.shape}")
print(f" ✅ Features shape: {features.shape}") print(f" ✅ Features shape: {features.shape}")
print(f" ✅ Multi-timeframe predictions shape: {multi_timeframe_pred.shape}")
else: else:
print(f" ✅ Output shape: {output.shape}") print(f" ✅ Output shape: {output.shape}")