test = isg.ISG3D_UAI(9, 9, 9)
elif test_name == 'rbm':
print 'Running RBM_UAI for {0} Iterations.'.format(num_iter)
test = isg.RBM_UAI()
elif test_name == 'ferrof':
print 'Running Ferro2D_Frustrated for {0} Iterations.'.format(num_iter)
test = isg.Ferro2D_Frustrated(60, 60)
elif test_name == 'chi':
print "Running Chimera."
test = isg.test = isg.Chimera(4,4,4)
temp = Ising_Gibbs(test)
import time
time_1 = time.time()
temp.gibbs_random(num_iter)
time_2 = time.time()
print "Sampling lasted: ", time_2 - time_1
print 'Saving results in ' + './Regular_Gibbs_' + test_name + '_' + str(num_iter) + '.npy'
cur_res_num = ASAW_config.get_and_inc_cur_res_num('path_config.cfg')
f = open('Regular_Gibbs_' + test_name + '_' + str(num_iter) + '.npy', 'w')
save(f, temp.energies[1:])
f.close()
temp.compute_acf(max_lag = min(num_iter-1, 500), subsample = max(num_iter/100000, 1))
temp.plots()
num_iter = int(sys.argv[1])
elif len(sys.argv) > 1:
num_iter = int(sys.argv[1])
if batch:
rbm = isg.RBM_UAI_BLOCK()
v = zeros(784)
total_energy = []
for i in xrange(5):
v[0:428] = ones(428)
bbs = Block_RBM_Sampler(rbm, v, None)
total_energy.append(bbs.iterate(num_iter))
cur_res_num = ASAW_config.get_and_inc_cur_res_num("path_config.cfg")
bbs.acf_plot_multi_energies(total_energy, res_num=cur_res_num)
f = open("Block_RBM_" + str(cur_res_num) + ".npy", "w")
save(f, total_energy)
f.close()
else:
rbm = isg.RBM_UAI_BLOCK()
v = zeros(784)
total_energy = []
v[0:428] = ones(428)
bbs = Block_RBM_Sampler(rbm, v, None)
import time
time_1 = time.time()
