Skip to content

Commit

Permalink
Fix input data reshaping to match MulticoreBPFLayer expectations
Browse files Browse the repository at this point in the history
  • Loading branch information
devin-ai-integration[bot] committed Jul 9, 2024
1 parent aa546af commit d394ec6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/deep_learning_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -338,8 +338,8 @@ def predict(model, data):
forward_matrix = np.random.rand(100, 3) # Example shape (num_particles, 3)

# Create the deep learning model with dynamic input shape
input_shape = (1, data.shape[1]) # Ensure input shape is 2D: (1, features)
data = data.reshape((data.shape[0], 1, data.shape[1])) # Reshape data to (samples, 1, features)
input_shape = (data.shape[1], data.shape[2]) # Ensure input shape is 3D: (timesteps, features)
data = data.reshape((data.shape[0], data.shape[1], data.shape[2])) # Reshape data to (samples, timesteps, features)
model = create_deep_learning_model(input_shape, transition_matrix, process_noise_cov, forward_matrix)
model.summary()

Expand Down

0 comments on commit d394ec6

Please sign in to comment.