From 228f7f626dddcff81d95923ad18d1a1002be1912 Mon Sep 17 00:00:00 2001 From: Simone Carlo Surace Date: Tue, 3 Oct 2023 22:48:12 +0200 Subject: [PATCH] Make space-time examples smaller --- examples/approx_space_time_learning.jl | 2 +- examples/exact_space_time_learning.jl | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/approx_space_time_learning.jl b/examples/approx_space_time_learning.jl index eeb68f6a..5052ad32 100644 --- a/examples/approx_space_time_learning.jl +++ b/examples/approx_space_time_learning.jl @@ -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); diff --git a/examples/exact_space_time_learning.jl b/examples/exact_space_time_learning.jl index 7971037a..58584aaa 100644 --- a/examples/exact_space_time_learning.jl +++ b/examples/exact_space_time_learning.jl @@ -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); @@ -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);