We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent aa949e6 commit 15eaec8Copy full SHA for 15eaec8
visualize_mean_var.py
@@ -22,7 +22,7 @@ def draw_distributions(filename, save_dir, type='mean', node_no=0, save_plots=Fa
22
for i in range(len(means)):
23
mean = np.mean(means[i].reshape((batch_size, -1))[:, node_no])
24
std = np.sum(np.square(stds[i].reshape((batch_size, -1))[:, node_no])) / batch_size
25
- sns.distplot(np.random.normal(loc=mean, scale=std, size=1000), ax=ax)
+ sns.distplot(np.random.normal(loc=mean, scale=std, size=1000), ax=ax, hist=False)
26
ax.axvline(mean, color='r', linestyle='-')
27
iteration = i % freq
28
epoch = i // freq
0 commit comments