Included missing weights when calculating inner products

This commit is contained in:
Nuwan Yapa 2024-06-28 17:00:51 -04:00
parent 5fc9d8022e
commit 6d254480fe
1 changed files with 4 additions and 3 deletions

View File

@ -28,6 +28,7 @@ js = collect(Iterators.filter(tri, iter_prod(0:jmax, 0:jmax)))
basis = iter_prod(js, zip(ks, ws), zip(ks, ws)) # basis = ((j1, j2), (k1, w1), (k2, w2))
basis_size = length(js) * length(ks)^2
weights_mat = spdiagm(repeat(kron(ws, ws), jmax + 1))
@assert length(basis) == basis_size "Something wrong with the basis"
println("Basis size = $basis_size")
@ -96,9 +97,9 @@ end
training_vecs = vcat(training_vecs, conj(training_vecs))
EC_basis = hcat(training_vecs...)
N_EC = transpose(EC_basis) * EC_basis
Ha_EC = transpose(EC_basis) * Ha * EC_basis
Vb_EC = transpose(EC_basis) * Vb * EC_basis
N_EC = transpose(EC_basis) * weights_mat * EC_basis
Ha_EC = transpose(EC_basis) * weights_mat * Ha * EC_basis
Vb_EC = transpose(EC_basis) * weights_mat * Vb * EC_basis
current_E = 4.0766890719636635 - 0.01275892774109674im