Skip to content

Commit

Permalink
Make space-time examples smaller
Browse files Browse the repository at this point in the history
  • Loading branch information
simsurace committed Oct 3, 2023
1 parent 23a9638 commit 228f7f6
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion examples/approx_space_time_learning.jl
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ end
# Also works with RectilinearGrids of inputs.
# Times must be increasing, points in space can be anywhere.
N = 50;
T = 1000;
T = 100;
points_in_space = [randn(N) for _ in 1:T];
points_in_time = RegularSpacing(0.0, 0.1, T);
x = RegularInTime(points_in_time, points_in_space);
Expand Down
4 changes: 2 additions & 2 deletions examples/exact_space_time_learning.jl
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ end
# Exact inference only works for such grids.
# Times must be increasing, points in space can be anywhere.
N = 50;
T = 1_000;
T = 100;
points_in_space = collect(range(-3.0, 3.0; length=N));
points_in_time = RegularSpacing(0.0, 0.01, T);
x = RectilinearGrid(points_in_space, points_in_time);
Expand Down Expand Up @@ -73,7 +73,7 @@ final_params = unpack(training_results.minimizer)
f_post = posterior(build_gp(final_params)(x, final_params.var_noise), y);

# Specify some locations at which to make predictions.
T_pr = 1200;
T_pr = 120;
points_in_time_pr = RegularSpacing(0.0, 0.01, T_pr);
x_pr = RectilinearGrid(points_in_space, points_in_time_pr);

Expand Down

0 comments on commit 228f7f6

Please sign in to comment.