Newbeeer/pfgmpp

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)