def dmrgrun(self,model):
'''Get the result from DMRG'''
#run dmrg to get the initial guess.
hgen=ExpandGenerator(spaceconfig=model.spaceconfig,H=model.H_serial,evolutor_type='masked')
dmrgegn=DMRGEngine(hgen=hgen,tol=0,reflect=True)
dmrgegn.use_U1_symmetry('M',target_block=zeros(1))
EG,mps=dmrgegn.run_finite(endpoint=(5,'<-',0),maxN=30,tol=1e-12)
return EG,mps
def test_dmrg_infinite(self):
'''test for infinite dmrg.'''
maxiter=100
model=self.get_model(maxiter+2,1)
hgen=RGHGen(spaceconfig=SpinSpaceConfig([2,1]),H=model.H_serial,evolutor_type='masked')
dmrgegn=DMRGEngine(hgen=hgen,tol=0,reflect=True)
dmrgegn.use_U1_symmetry('M',target_block=0)
EG=dmrgegn.run_infinite(maxiter=maxiter,maxN=20,tol=0)[0]
assert_almost_equal(EG,0.25-log(2),decimal=2)
def test_dmrg_infinite(self):
'''test for infinite dmrg.'''
maxiter = 100
model = self.get_model(maxiter + 2, 1)
hgen = ExpandGenerator(spaceconfig=SpinSpaceConfig([1, 2]),
H=model.H_serial,
evolutor_type='masked')
dmrgegn = DMRGEngine(hgen=hgen, tol=0, reflect=True, iprint=10)
dmrgegn.use_U1_symmetry('M', target_block=zeros(1))
EG = dmrgegn.run_infinite(maxiter=maxiter, maxN=20, tol=0)[0]
assert_almost_equal(EG, 0.25 - log(2), decimal=2)
def test_dmrg_finite(self):
'''
Run iDMFT for heisenberg model.
'''
nsite=10
model=self.get_model(nsite,1)
hgen1=RGHGen(spaceconfig=SpinSpaceConfig([2,1]),H=model.H_serial,evolutor_type='null')
hgen2=RGHGen(spaceconfig=SpinSpaceConfig([2,1]),H=model.H_serial,evolutor_type='masked')
H=get_H(hgen1)
EG1=eigsh(H,k=1,which='SA')[0]
dmrgegn=DMRGEngine(hgen=hgen2,tol=0,reflect=True)
dmrgegn.use_U1_symmetry('M',target_block=0)
EG2=dmrgegn.run_finite(endpoint=(5,'<-',0),maxN=[10,20,30,40,40],tol=0)[0]
assert_almost_equal(EG1,EG2,decimal=4)
def spin1_dmrg(J, Jz, h, nsite=10, which="finite", J2=0, J2z=None):
"""
Run iDMFT for heisenberg model.
"""
if J2 == 0:
model = Spin1Model(J=J, Jz=Jz, h=h, nsite=nsite)
else:
model = Spin1Model(J=J, Jz=Jz, J2=J2, J2z=J2z, h=h, nsite=nsite)
hgen = RGHGen(spaceconfig=SpinSpaceConfig([2, 1]), H=model.H_serial, evolutor_type="masked")
dmrgegn = DMRGEngine(hgen=hgen, tol=0, reflect=True)
dmrgegn.use_U1_symmetry("M", target_block=0)
if which == "infinite":
dmrgegn.run_infinite(maxiter=50, maxN=20, tol=0)
elif which == "lanczos":
dmrgegn.direct_solve()
else:
dmrgegn.run_finite(endpoint=(5, "<-", 0), maxN=[10, 20, 30, 40, 40], tol=0)
pdb.set_trace()
def test_vmps(self):
'''
Run vMPS for Heisenberg model.
'''
nsite=10
filename='mps_heisenberg_%s.dat'%(nsite)
model=self.get_model(nsite,which=1)
#mps=MPS.load(filename)
#run dmrg to get the initial guess.
hgen=RGHGen(spaceconfig=SpinSpaceConfig([2,1]),H=model.H_serial,evolutor_type='masked')
dmrgegn=DMRGEngine(hgen=hgen,tol=0,reflect=True)
dmrgegn.use_U1_symmetry('M',target_block=0)
EG,mps=dmrgegn.run_finite(endpoint=(1,'<-',0),maxN=40,tol=1e-12)
mps.save(filename)
#run vmps
vegn=VMPSEngine(H=model.H,k0=mps)
vegn.run()
def test_dmrg_finite(self):
'''
Run iDMFT for heisenberg model.
'''
nsite = 10
model = self.get_model(nsite, 1)
hgen1 = ExpandGenerator(spaceconfig=SpinSpaceConfig([1, 2]),
H=model.H_serial,
evolutor_type='null')
hgen2 = ExpandGenerator(spaceconfig=SpinSpaceConfig([1, 2]),
H=model.H_serial,
evolutor_type='masked')
H = get_H(hgen1)
EG1 = eigsh(H, k=1, which='SA')[0]
dmrgegn = DMRGEngine(hgen=hgen2, tol=0, reflect=True)
dmrgegn.use_U1_symmetry('M', target_block=zeros(1))
EG2 = dmrgegn.run_finite(endpoint=(5, '<-', 0),
maxN=[10, 20, 40, 40, 40],
tol=0)[0]
assert_almost_equal(EG1, EG2, decimal=4)
def solve1(J,Jz,h,target_block,nsite=10,parity=None,nspin=2):
'''
Run iDMFT for heisenberg model.
'''
model=HeisenbergModel(J=J,Jz=Jz,h=h,nsite=nsite,nspin=nspin)
suffix='%s_%s%s.dat'%(model.get_str(),target_block,parity)
filename_mps='data/mps_'+suffix
filename_eng='data/eng_'+suffix
parity=None
hgen=RGHGen(spaceconfig=model.spaceconfig,H=model.H_serial,evolutor_type='masked')
dmrgegn=DMRGEngine(hgen=hgen,tol=0,reflect=parity is not None,eigen_solver='JD')
dmrgegn.use_U1_symmetry('M',target_block=target_block)
if parity is not None:
dmrgegn.use_disc_symmetry({'C':parity})
t0=time.time()
Emin,mps=dmrgegn.run_finite(endpoint=(5,'<-',0),maxN=[10,20,40,80,80],tol=1e-8)
t1=time.time()
print 'The Ground State Energy for %s: %s, Elapse %s.'%(model,Emin,t1-t0)
print 'FIN - CORE:%s'%RANK
mps.save(filename_mps)
savetxt(filename_eng,Emin)
return Emin,mps
