Target one resonance
This commit is contained in:
parent
49678b4404
commit
d480971eea
|
|
@ -1,8 +1,11 @@
|
||||||
|
using Plots
|
||||||
|
|
||||||
include("../EC.jl")
|
include("../EC.jl")
|
||||||
include("../ho_basis.jl")
|
include("../ho_basis.jl")
|
||||||
include("../p_space.jl")
|
include("../p_space.jl")
|
||||||
|
|
||||||
μω_gen = 0.5 * exp(-1im * 0.47 * pi)
|
angle = 0.47 * pi
|
||||||
|
μω_gen = 0.5 * exp(-1im * angle)
|
||||||
μ = 0.5
|
μ = 0.5
|
||||||
l = 0
|
l = 0
|
||||||
V1 = -5
|
V1 = -5
|
||||||
|
|
@ -19,12 +22,18 @@ V = V1 .* V_Gaussian.(R1, l, ns, transpose(ns); μω_gen=μω_gen) + V2 .* V_Gau
|
||||||
|
|
||||||
n_EC = 8
|
n_EC = 8
|
||||||
train_cs = (0.7 .+ 0.05 * randn(n_EC)) - 1im * (0.2 .+ 0.05 * randn(n_EC))
|
train_cs = (0.7 .+ 0.05 * randn(n_EC)) - 1im * (0.2 .+ 0.05 * randn(n_EC))
|
||||||
target_cs = range(0.77, 0.22, 6)
|
target_cs = [0.5]
|
||||||
|
|
||||||
near_E = 0.2 + 0.2im
|
near_E = 0.2 + 0.2im
|
||||||
|
exact_E = [0.20845136860234303 - 0.07100640993695649im]
|
||||||
|
|
||||||
EC = affine_EC(T, V)
|
EC = affine_EC(T, V; ensemble_size=32)
|
||||||
train!(EC, train_cs; ref_eval=near_E, CAEC=false)
|
train!(EC, train_cs; ref_eval=near_E, CAEC=false)
|
||||||
extrapolate!(EC, target_cs)
|
extrapolate!(EC, target_cs; precalculated_exact_E=exact_E)
|
||||||
|
|
||||||
plot(EC, "temp/XZ.pdf"; xlims=(-0.2,0.3), ylims=(-0.3,0.3))
|
plot(EC; xlims=(0,0.3), ylims=(-0.3,0.3))
|
||||||
|
hline!([0], color=:red, label="continuum")
|
||||||
|
xlabel!("Re(E)")
|
||||||
|
ylabel!("Im(E)")
|
||||||
|
plot!(legend=:bottomleft)
|
||||||
|
savefig("temp/XZ.pdf")
|
||||||
Loading…
Reference in New Issue