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

Verbose bugs fixed

This commit is contained in:
Nuwan Yapa 2025-01-15 17:38:01 -05:00
parent 870eecbb38
commit ad666a41d0
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
EC.jl
View File

@ -72,8 +72,6 @@ function extrapolate!(EC::affine_EC, c_vals; ref_eval=EC.training_E[end], verbos
@assert EC.trained "EC model must be trained using train() before extrapolation" @assert EC.trained "EC model must be trained using train() before extrapolation"
for c in c_vals for c in c_vals
verbose && println("Extact solution for c = $c")
global current_E global current_E
if isnothing(precalculated_exact_E) if isnothing(precalculated_exact_E)

8
ho_basis.jl
View File

@ -248,15 +248,15 @@ function get_3b_H_matrix(coord_system::coordinate_system, V_of_r, μω_global, E
μ1 = μ2 = μ = m/2 μ1 = μ2 = μ = m/2
end end
verbose || println("No of threads = ", Threads.nthreads()) verbose && println("No of threads = ", Threads.nthreads())
@time "Basis" basis = ho_basis_2B(E_max, Λ) @time "Basis" basis = ho_basis_2B(E_max, Λ)
verbose || println("Basis size = ", basis.dim) verbose && println("Basis size = ", basis.dim)
out = spzeros(basis.dim, basis.dim) out = spzeros(basis.dim, basis.dim)
if kinetic_part if kinetic_part
verbose || println("Constructing KE matrices") verbose && println("Constructing KE matrices")
@time "T1" out += get_sp_T_matrix(basis.n1s, basis.l1s, [basis.n2s, basis.l2s]; μω_gen=μ1ω1, μ=μ1) @time "T1" out += get_sp_T_matrix(basis.n1s, basis.l1s, [basis.n2s, basis.l2s]; μω_gen=μ1ω1, μ=μ1)
@time "T2" out += get_sp_T_matrix(basis.n2s, basis.l2s, [basis.n1s, basis.l1s]; μω_gen=μ2ω2, μ=μ2) @time "T2" out += get_sp_T_matrix(basis.n2s, basis.l2s, [basis.n1s, basis.l1s]; μω_gen=μ2ω2, μ=μ2)
if coord_system == src if coord_system == src
@ -265,7 +265,7 @@ function get_3b_H_matrix(coord_system::coordinate_system, V_of_r, μω_global, E
end end
if potential_part if potential_part
verbose || println("Constructing PE matrices") verbose && println("Constructing PE matrices")
if coord_system == jacobi if coord_system == jacobi
@time "V" out += get_jacobi_V_matrix(V_of_r, basis, μ1ω1, μω_global; atol=atol, maxevals=maxevals) @time "V" out += get_jacobi_V_matrix(V_of_r, basis, μ1ω1, μω_global; atol=atol, maxevals=maxevals)
elseif coord_system == src elseif coord_system == src