edit test
This commit is contained in:
@ -46,12 +46,13 @@ def test_dqn_architecture():
|
||||
output = network(test_input)
|
||||
|
||||
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" ✅ Regime prediction shape: {regime_pred.shape}")
|
||||
print(f" ✅ Price prediction shape: {price_pred.shape}")
|
||||
print(f" ✅ Volatility prediction shape: {volatility_pred.shape}")
|
||||
print(f" ✅ Features shape: {features.shape}")
|
||||
print(f" ✅ Multi-timeframe predictions shape: {multi_timeframe_pred.shape}")
|
||||
else:
|
||||
print(f" ✅ Output shape: {output.shape}")
|
||||
|
||||
|
Reference in New Issue
Block a user