Update 18_2_1D_Diffusion_Model.ipynb

This commit is contained in:
Pietro Monticone
2023-10-30 18:17:44 +01:00
parent aa5c7f77d6
commit f855263f9f

View File

@@ -172,7 +172,7 @@
" # Find diffusion kernel for every x_train and draw samples\n", " # Find diffusion kernel for every x_train and draw samples\n",
" dk_mean, dk_std = diffusion_kernel(x_train, t, beta)\n", " dk_mean, dk_std = diffusion_kernel(x_train, t, beta)\n",
" z_t = np.random.normal(size=x_train.shape) * dk_std + dk_mean\n", " z_t = np.random.normal(size=x_train.shape) * dk_std + dk_mean\n",
" # Find conditional diffusion distribution for each x_train, z pair and draw samlpes\n", " # Find conditional diffusion distribution for each x_train, z pair and draw samples\n",
" cd_mean, cd_std = conditional_diffusion_distribution(x_train,z_t,t,beta)\n", " cd_mean, cd_std = conditional_diffusion_distribution(x_train,z_t,t,beta)\n",
" if t == 1:\n", " if t == 1:\n",
" z_tminus1 = x_train\n", " z_tminus1 = x_train\n",