Save data

This commit is contained in:
Nuwan Yapa 2024-07-09 15:14:55 -04:00
parent 147b980b06
commit cac2733bb8
1 changed files with 11 additions and 2 deletions

View File

@ -1,4 +1,4 @@
using Plots using Plots, DelimitedFiles
include("../p_space.jl") include("../p_space.jl")
# contour # contour
@ -51,10 +51,19 @@ for (j, c) in enumerate(extrapolate_points)
extrapolate_E[j] = evals[i] extrapolate_E[j] = evals[i]
end end
open("temp/2b_GSM.csv", "w") do f
write(f, "# training \n")
writedlm(f, hcat(reim(training_E)...))
write(f, "# exact \n")
writedlm(f, hcat(reim(exact_E)...))
write(f, "# extrapolate \n")
writedlm(f, hcat(reim(extrapolate_E)...))
end
scatter(real.(training_E), imag.(training_E), label="training") scatter(real.(training_E), imag.(training_E), label="training")
scatter!(real.(exact_E), imag.(exact_E), label="exact") scatter!(real.(exact_E), imag.(exact_E), label="exact")
scatter!(real.(extrapolate_E), imag.(extrapolate_E), label="extrapolated") scatter!(real.(extrapolate_E), imag.(extrapolate_E), label="extrapolated")
scatter!(real.(basis_E), imag.(basis_E), m=:x, label="Berggren basis") scatter!(real.(basis_E), imag.(basis_E), m=:x, label="Berggren basis")
xlims!(-0.3,0.3) xlims!(-0.3,0.3)
ylims!(-0.120,0.020) ylims!(-0.120,0.020)
savefig("temp/" * string(rand(UInt16)) * ".pdf") savefig("temp/2b_GSM.pdf")