Skip to content

Commit f542fd9

Browse files
committed
Add placeholder for rollout
1 parent 9caedbc commit f542fd9

1 file changed

Lines changed: 23 additions & 0 deletions

File tree

autoemulate/experimental/emulators/spatiotemporal.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,3 +52,26 @@ def predict(
5252

5353
@abstractmethod
5454
def _predict(self, x: TensorLike | DataLoader, with_grad: bool) -> OutputLike: ...
55+
56+
# TODO: add method for rollout predictions
57+
# def predict_rollout(self, x: DataLoader, timesteps: int = 1) -> OutputLike:
58+
# """
59+
# Predict the output for the given input, rolling out for a number of timesteps.
60+
61+
# Parameters
62+
# ----------
63+
# x: DataLoader
64+
# Input `DataLoader` to make predictions for.
65+
# timesteps: int
66+
# Number of timesteps to rollout for. Defaults to 1.
67+
68+
# Returns
69+
# -------
70+
# OutputLike
71+
# The emulator predicted spatiotemporal output.
72+
# """
73+
74+
# # Start at t=0 x_0
75+
# # model predicts x_1 given x_0
76+
# # then model predicts x_2 given model's predicted x_1
77+
# # then model predicts x_3 given model's predicted x_2

0 commit comments

Comments
 (0)