Beginning Anomaly Detection Using

plot_training_loss: Whether or not to plot the training loss. Default is 

True.

Chapter 5   Boltzmann maChines

---

192

Now that you finished training, you can look at evaluating your model. To get the 

probability values for each entry in the test set, you have to calculate the free energy for 

each data point (this is a function unique to this version of the RBM). From there, you 

can get the probability of each data point occurring given its free energy. Run the code in 

Figure 

5-28

.

Now you can fit the data to the model. Run the following (refer to Figure 

5-27

 for the 

output):

model.fit(x_train, validation_set=x_test)