31 lines
714 B
Julia
31 lines
714 B
Julia
include("../EC.jl")
|
|
include("../ho_basis.jl")
|
|
include("../p_space.jl")
|
|
|
|
μω_gen = 0.5 * exp(-1im * 0.47 * pi)
|
|
μ = 0.5
|
|
l = 0
|
|
V1 = -5
|
|
R1 = sqrt(3)
|
|
V2 = 2
|
|
R2 = sqrt(10)
|
|
n_max = 15
|
|
|
|
ns = collect(0:n_max)
|
|
ls = fill(l, n_max + 1)
|
|
|
|
T = get_sp_T_matrix(ns, ls; μω_gen=μω_gen, μ=μ)
|
|
V = V1 .* V_Gaussian.(R1, l, ns, transpose(ns); μω_gen=μω_gen) + V2 .* V_Gaussian.(R2, l, ns, transpose(ns); μω_gen=μω_gen)
|
|
|
|
n_EC = 8
|
|
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)
|
|
|
|
near_E = 0.2 + 0.2im
|
|
|
|
EC = affine_EC(T, V)
|
|
train!(EC, train_cs; ref_eval=near_E, CAEC=false)
|
|
extrapolate!(EC, target_cs)
|
|
|
|
plot(EC, "temp/XZ.pdf"; xlims=(-0.2,0.3), ylims=(-0.3,0.3))
|