yapanuwan
/
BergEC-jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Plot formatting

This commit is contained in:
Nuwan Yapa 2025-06-11 13:48:32 -04:00
parent d158d022dc
commit 49678b4404
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
EC.jl
View File

@ -156,11 +156,11 @@ exportCSV(EC::affine_EC, filename) = exportCSV(filename, (EC.training_E, EC.exac
"Plot EC data and optionally save figure to a file" "Plot EC data and optionally save figure to a file"
function plot(EC::affine_EC, save_fig_filename=nothing; basis_points=nothing, basis_contour=nothing, xlims=nothing, ylims=nothing) function plot(EC::affine_EC, save_fig_filename=nothing; basis_points=nothing, basis_contour=nothing, xlims=nothing, ylims=nothing)
scatter(real.(EC.training_E), imag.(EC.training_E), label="training") scatter(real.(EC.training_E), imag.(EC.training_E), label="training")
scatter!(real.(EC.exact_E), imag.(EC.exact_E), label="exact") scatter!(real.(EC.exact_E), imag.(EC.exact_E), label="exact", markercolor=:white)
if EC.ensemble_size > 0 if EC.ensemble_size > 0
scatter!(real.(EC.extrapolated_E), imag.(EC.extrapolated_E), xerror=real.(EC.extrapolated_CI), yerror=imag.(EC.extrapolated_CI), label="extrapolated") scatter!(real.(EC.extrapolated_E), imag.(EC.extrapolated_E), xerror=real.(EC.extrapolated_CI), yerror=imag.(EC.extrapolated_CI), label="extrapolated", m=:x)
else else
scatter!(real.(EC.extrapolated_E), imag.(EC.extrapolated_E), label="extrapolated") scatter!(real.(EC.extrapolated_E), imag.(EC.extrapolated_E), label="extrapolated", m=:x)
end end
isnothing(basis_points) || scatter!(real.(basis_points), imag.(basis_points), m=:x, label="basis") isnothing(basis_points) || scatter!(real.(basis_points), imag.(basis_points), m=:x, label="basis")