Add dns proj vs noproj plot

Author: SCiarella

benchmark.jl

@@ -111,61 +111,69 @@ plot_labels = Dict(
     #    xlabel = "Iteration",
     #    ylabel = "DCF",
     #),
-    :divergence => (
-        title  = "Divergence for different configurations",
+    :dns_solution => (
+        title  = "DNS solution for different configurations",
         xlabel = "t",
-        ylabel = "Face-average",
-    ),
-    :energy_evolution => (
-        title = "Energy evolution for different configurations",
-        xlabel = "t",
-        ylabel = "E(t)",
-    ),
-    :energy_evolution_hist => (
-        title = "Energy histogram for different configurations",
-        xlabel = "frequency",
-        ylabel = "E(t)",
-    ),
-    :energy_spectra => (
-        title  = "Energy spectra",
-    ),
-    :training_time => (
-        title  = "Training time for different configurations",
-        xlabel = "Model",
-        ylabel = "Training time (s) (per iteration)",
-    ),
-    :training_comptime => (
-        title  = "Training time for different configurations",
-        xlabel = "Model",
-        ylabel = "Full Training time (s)",
-    ),
-    :inference_time => (
-        title  = "Inference time for different configurations",
-        xlabel = "Model",
-        ylabel = "Inference time (s)",
-    ),
-    :num_parameters => (
-        title  = "Number of parameters for different configurations",
-        xlabel = "Model",
-        ylabel = "Number of parameters",
-    ),
-    :eprior => (
-        title  = "A-prior error for different configurations",
-        xlabel = "Model",
-        ylabel = "A-prior error",
-    ),
-    :epost => (
-        title  = "A-posteriori error for different configurations",
-        xlabel = "Model",
-        ylabel = "A-posteriori error",
-    ),
-    :epost_vs_t => (
-        title = "A-posteriori error as a function of time",
-        xlabel = "t",
-        ylabel = L"e_{M}(t)",
+        ylabel = "MAE",
     ),
+    #:divergence => (
+    #    title  = "Divergence for different configurations",
+    #    xlabel = "t",
+    #    ylabel = "Face-average",
+    #),
+    #:energy_evolution => (
+    #    title = "Energy evolution for different configurations",
+    #    xlabel = "t",
+    #    ylabel = "E(t)",
+    #),
+    #:energy_evolution_hist => (
+    #    title = "Energy histogram for different configurations",
+    #    xlabel = "frequency",
+    #    ylabel = "E(t)",
+    #),
+    #:energy_spectra => (
+    #    title  = "Energy spectra",
+    #),
+    #:training_time => (
+    #    title  = "Training time for different configurations",
+    #    xlabel = "Model",
+    #    ylabel = "Training time (s) (per iteration)",
+    #),
+    #:training_comptime => (
+    #    title  = "Training time for different configurations",
+    #    xlabel = "Model",
+    #    ylabel = "Full Training time (s)",
+    #),
+    #:inference_time => (
+    #    title  = "Inference time for different configurations",
+    #    xlabel = "Model",
+    #    ylabel = "Inference time (s)",
+    #),
+    #:num_parameters => (
+    #    title  = "Number of parameters for different configurations",
+    #    xlabel = "Model",
+    #    ylabel = "Number of parameters",
+    #),
+    #:eprior => (
+    #    title  = "A-prior error for different configurations",
+    #    xlabel = "Model",
+    #    ylabel = "A-prior error",
+    #),
+    #:epost => (
+    #    title  = "A-posteriori error for different configurations",
+    #    xlabel = "Model",
+    #    ylabel = "A-posteriori error",
+    #),
+    #:epost_vs_t => (
+    #    title = "A-posteriori error as a function of time",
+    #    xlabel = "t",
+    #    ylabel = L"e_{M}(t)",
+    #),
 )
 
+dns_seeds = splitseed(123456, 8)
+dns_seeds = [0x185efb6b]
+
 for key in keys(plot_labels)
     @info "Plotting $key"
 
@@ -305,12 +313,20 @@ for key in keys(plot_labels)
                     plot_epost_vs_t(
                         error_file, closure_name, nles, ax, color, PLOT_STYLES
                     )
+                elseif key == :dns_solution
+                    data_ref = load_data_set(outdir, nles, Φ, dns_seeds, false)
+                    data_proj = load_data_set(outdir, nles, Φ, dns_seeds, true)
+                    plot_dns_solution(
+                        data_ref, data_proj, ax#, color, PLOT_STYLES
+                    )
+                else
+                    @error "Unknown plot type: $key"
                 end
             end
         end
     end
     # Add legend
-    if key != :energy_spectra
+    if !(key in (:energy_spectra, :dns_solution))
         Legend(fig[:, end+1], ax)
     end
 
@@ -320,7 +336,7 @@ for key in keys(plot_labels)
     end
 
     # Set log-log scale
-    if key == :epost_vs_t
+    if key in (:epost_vs_t, :dns_solution)
         ax.xscale = log10
         ax.yscale = log10
     end

configs/local/cnn_proj.yaml

@@ -10,8 +10,8 @@ params:
   tburn: 0.5
   tsim: 2.0
   savefreq: 50
-  ndns: 1024
-  nles: [64]
+  ndns: 128
+  nles: [32]
   filters: ["FaceAverage()"]
   icfunc: "(setup, psolver, rng) -> random_field(setup, T(0); kp=20, psolver, rng)"
   method: "RKMethods.Wray3(; T)"

src/Benchmark.jl

@@ -110,9 +110,11 @@ export _convert_to_single_index,
     plot_inference_time,
     plot_num_parameters,
     plot_error,
-    plot_epost_vs_t
+    plot_epost_vs_t,
+    plot_dns_solution
 
 export compute_eprior, compute_epost, compute_t_prior_inference
 export reusepriorfile, reusepostfile
+export load_data_set
 
 end # module Benchmark

src/plots.jl

@@ -886,3 +886,22 @@ function plot_epost_vs_t(error_file, closure_name, nles, ax, color, PLOT_STYLES)
 
     return nothing
 end
+
+function plot_dns_solution(data_ref, data_proj, ax) 
+
+    for (i,dr) in enumerate(data_ref)
+        x = dr.t[2:end] #skip t=0
+        y = []
+        for (ti, tval) in enumerate(x)
+            #@info "t(ref) = $(tval)   t(proj)=$(data_proj[i].t[ti])"
+            #@assert tval ≈ data_proj[i].t[ti]
+            push!(y, sum(abs, dr.u[:,:,:,ti]-data_proj[i].u[:,:,:,ti]))
+        end
+        scatterlines!(
+            ax,
+            x,
+            vec(y)
+        )
+    end
+    
+end