Skip to content

Commit

Permalink
add main files
Browse files Browse the repository at this point in the history
  • Loading branch information
Joel Bierman committed Dec 5, 2023
1 parent 66fb0d2 commit 4a87f7c
Show file tree
Hide file tree
Showing 36 changed files with 6,221 additions and 19 deletions.
1 change: 1 addition & 0 deletions electronic_structure_algorithms/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@

Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
from .ssvqe import SSVQE, SSVQEResult
from .vqd import VQD, VQDResult
from .mcvqe import MCVQE, MCVQEResult
524 changes: 524 additions & 0 deletions electronic_structure_algorithms/excited_states_eigensolvers/mcvqe.py

Large diffs are not rendered by default.

627 changes: 627 additions & 0 deletions electronic_structure_algorithms/excited_states_eigensolvers/ssvqe.py

Large diffs are not rendered by default.

537 changes: 537 additions & 0 deletions electronic_structure_algorithms/excited_states_eigensolvers/vqd.py

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
import torch

def get_HF_permutation_matrix(num_original_spin_orbitals: int,
num_spin_orbitals: int):

num_original_molecular_orbitals = int(num_original_spin_orbitals/2)
num_molecular_orbitals = int(num_spin_orbitals/2)

initial_partial_unitary_guess = torch.zeros(size=(num_original_molecular_orbitals, num_molecular_orbitals), dtype=torch.float64)
for n in range(int(num_molecular_orbitals)):
initial_partial_unitary_guess[n,n] = 1.0

return initial_partial_unitary_guess

2 changes: 2 additions & 0 deletions electronic_structure_algorithms/initializations/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from .configuration_interaction_states import get_CIS_states, get_CISD_states
from .HF_permutation_matrix import get_HF_permutation_matrix
Loading

0 comments on commit 4a87f7c

Please sign in to comment.