# # Sample Data Preparation
#     input_dim = 3  # Example input dimension
#     hidden_dim = 5  # Example hidden dimension
#     num_samples = 1000
#     sequence_length = 10

#     # Generate dummy data for training
#     X = np.random.randn(num_samples, input_dim, 1)  # Shape (num_samples, input_dim, 1)
#     y = np.random.randn(
#         num_samples, hidden_dim, 1
#     )  # Shape (num_samples, hidden_dim, 1)

#     # Initialize and train the LSTM
#     lstm = LSTM(input_dim, hidden_dim)
#     lstm.train(X, y, num_epochs=10, learning_rate=0.01)