Skip to content

Commit 15eaec8

Browse files
committedMar 2, 2020
distplot for both mean and var without histogram
1 parent aa949e6 commit 15eaec8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎visualize_mean_var.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def draw_distributions(filename, save_dir, type='mean', node_no=0, save_plots=Fa
2222
for i in range(len(means)):
2323
mean = np.mean(means[i].reshape((batch_size, -1))[:, node_no])
2424
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)
25+
sns.distplot(np.random.normal(loc=mean, scale=std, size=1000), ax=ax, hist=False)
2626
ax.axvline(mean, color='r', linestyle='-')
2727
iteration = i % freq
2828
epoch = i // freq

0 commit comments

Comments
 (0)