diff --git a/EC.jl b/EC.jl index b129bf7..b29f8ca 100644 --- a/EC.jl +++ b/EC.jl @@ -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" 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.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 - 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 - 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 isnothing(basis_points) || scatter!(real.(basis_points), imag.(basis_points), m=:x, label="basis")