f0a81e2d218aa41b538de5e5cfebcc149f758dc5,examples/rbm_tutorial.py,BinomialRBM,fit,#BinomialRBM#Any#Any#Any#Any#Any#Any#Any#,239

Before Change

            if ep % log_every == 0:

                fidelity_ = self.fidelity(vis, psi)
                print ("Epoch = ",ep,"\tFidelity = %.5f" %fidelity_.item())
                fidelity_list.append(fidelity_)
                epoch_list.append(ep)
           
            // Save fidelities at the end of training.

After Change

                plt.ylim([0.0,1.05])
                plt.xlabel("epoch",fontsize=20)
                //plt.ylabel("F",fontsize=20)
                plt.xticks(fontsize=20)
                plt.yticks(fontsize=20)
                plt.gca().text(0.95, 0.2, "F = "+"{:.3}".format(fidelity_.item()),
                        verticalalignment="bottom", horizontalalignment="right",
                        color="k", fontsize=20,transform=plt.gca().transAxes)

Instances