edit test
This commit is contained in:
@ -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}")
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user