Issues in pfgmpp_toy.ipynb
dbl001 opened this issue · 0 comments
dbl001 commented
Y not defined before use:
train_dataset = Gaussain_data(X=X, Y=Y)
Is this ok?
# Integer labels
Y = torch.cat([torch.zeros(size, dtype=torch.long),
torch.ones(size, dtype=torch.long)], dim=0)
Also, replace:
plt.savefig(title +'.png', bbox='tight', dpi=300)
with
plt.savefig(title +'.png', bbox_inches='tight', dpi=300)