<jupyter_start><jupyter_text>SGLD<jupyter_code>def sgld(X,Y,theta0,M,C,V,eps,nbatch,niter=2000):
    n,p=X.shape
    theta=theta0
    theta_save=np.zeros([niter,p])
    np.random.seed(10)
    for t in range(niter): 
        X_sample,Y_sample = batch(X,Y,nbatch)
        theta=theta-gradlogistic(X_sample,Y_sample,theta,n,V)*eps+np.random.multivariate_normal(np.zeros(p),np.sqrt(2*0.01*np.eye(p)),1).ravel()
        theta_save[t,:]=theta
    return theta_save
theta0 = np.zeros(5)
M = C = np.eye(5)
nbatch=500
eps=.001
V = np.diag([20,20,20,20,20])
points = np.arange(1,2010,100)
sgld_theta=sgld(X_train,Y_train,theta0,M,C,V,eps,nbatch,2000)
sglderr = test_err(sgld_theta)
plt.plot(points,sglderr)
plt.xlabel("iteration")
plt.ylabel("test error")
plt.title("SGLD")
plt.savefig("SGLD")<jupyter_output><empty_output><jupyter_text>SGHMC<jupyter_code>@jit
def batch_numba(X,Y, nbatch):
    nrow = X.shape[0]
    idx = np.random.choice(nrow, nbatch, replace = False)
    X_sample = X[idx,:]
    Y_sample = Y[idx]
    return X_sample, Y_sample
@jit([float64[:,:](float64[:],float64[:,:],float64[:],int64,float64[:],float64[:,:],float64[:,:],float64[:,:],float64,int64,int64)],cache = True)
def sghmc_numba(theta0,X,Y,nbatch,gradU,M,C,V,eps,step = 10, niter = 10):
    B = 1/2 * V * eps
    sigma = np.sqrt(2*eps*(C-B))
    n, p = X.shape
    theta = theta0 #set an initial value of theta
    thetas =np.zeros([step,p])
    Minv = np.linalg.inv(M)
    np.random.seed(10)
    #simulate dynamics
    for t in range(step):
        r = np.random.multivariate_normal(np.zeros(p),np.sqrt(M))
        for i in range(niter):
            theta = theta + eps*Minv@r
            X_sample,Y_sample = batch_numba(X,Y,nbatch)
            r =  r - eps*gradU(X_sample, Y_sample,theta,n,V) - eps*C @ Minv @ r 
            + np.random.multivariate_normal(np.zeros(p),sigma,1).ravel()
        thetas[t,:] = theta
    return thetas
theta0 = np.zeros(5)
M = C = np.eye(5)
nbatch=500
eps=.001
V = np.diag([20,20,20,20,20])
sghmc_theta = sghmc_numba(np.zeros(5),X_train,Y_train,nbatch,gradlogistic,M,C,V,eps,2000,50)
sghmcerr = test_err(sghmc_theta)
plt.plot(points,sghmcerr)
plt.xlabel("iteration")
plt.ylabel("test error")
plt.title("SGHMC")
plt.savefig("SGHMC")<jupyter_output><empty_output><jupyter_text>SGD (simplied version, without momentum)<jupyter_code>def gradlogistic_1(x,y,theta,n,V):  
    #theta is updated
    #X,Y are samples from the minibatch
    #n is total number of observations
    #V should be provided ahead, specified here
    Vinv = np.linalg.inv(V)
    d1 = -np.diag([(y-1/(1+np.exp(-x@theta)))])@x[None,:]
    d2 = Vinv@theta
    return d1 + d2
def sgd(X_train,y_train,nbatch,learning_rate,theta0, ninte):
    theta = theta0
    n,p = X_train.shape
    thetas = np.zeros([ninte,p])
    for i in range(ninte):
        grad = gradlogistic_1(X_train[i,:],y_train[i],theta0,1,V)
        theta = theta - learning_rate * grad
        thetas[i] = theta
    return thetas
theta0 = np.zeros(5)
M = C = np.eye(5)
nbatch=500
eps=.001
V = np.diag([20,20,20,20,20])
sgd_theta = sgd(X_train,Y_train,500,0.05,theta0,2000)
sgderr = test_err(sgd_theta)
plt.plot(points, sgderr)
plt.xlabel("iteration")
plt.ylabel("test error")
plt.title("SGD")
plt.savefig("SGD")<jupyter_output><empty_output><jupyter_text>Compare<jupyter_code>plt.plot(points,sglderr,"d-")
plt.plot(points,sghmcerr,color = "red")
plt.plot(points, sgderr, '--',color = "orange")
plt.legend(['SGLD','SGHMC','SGD'])
plt.xlabel("iteration")
plt.ylabel("test error")
plt.title("Comparison")
plt.savefig("Comparison")
plt.plot(points,sglderr,"d-")
plt.plot(points,sghmcerr,color = "red")
plt.legend(['SGLD','SGHMC'])
plt.xlabel("iteration")
plt.ylabel("test error")
plt.title("SGLD vs. SGHMC")
plt.savefig("SGLDvsSGHMC")<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.1798997625956212, 'fraction_numerical': 0.039567396465312585, 'mean_word_length': 4.851851851851852, 'pattern_counts': {'":': 0, '<': 17, '<?xml version=': 0, '>': 17, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '4', 'dir': 'jupyter-structured-clean-dedup', 'id': '581443', 'license': 'MIT', 'n_tokens_mistral': 1771, 'n_tokens_neox': 1635, 'n_words': 232, 'path': 'Comparison/Figure 4 Official.ipynb', 'repo_name': 'QianyinLu/SGHMC', 'size': '3791'}