vector predictions inference fix
This commit is contained in:
@ -7266,7 +7266,7 @@ class CleanTradingDashboard:
|
||||
|
||||
# Get prediction from CNN model
|
||||
with torch.no_grad():
|
||||
q_values, extrema_pred, price_pred, features_refined, advanced_pred = self.cnn_adapter(features_tensor)
|
||||
q_values, extrema_pred, price_pred, features_refined, advanced_pred, multi_timeframe_pred = self.cnn_adapter(features_tensor)
|
||||
|
||||
# Convert to probabilities using softmax
|
||||
action_probs = torch.softmax(q_values, dim=1)
|
||||
|
Reference in New Issue
Block a user