sphinxteam / Boltzmann.jl

Restricted Boltzmann Machines in Julia
Other
15 stars 4 forks source link

Validation Set #12

Closed eric-tramel closed 8 years ago

eric-tramel commented 8 years ago

In order to track the generality of the RBM during training, and to potentially perform cross-validation for on-line parameter updates, it would be nice to be able to pass a validation dataset to the fit() function as an optional keyword.

If the argument is passed, then the fit function should perform, potentially, per-epoch checks on the validation set (e.g. pseudo-likelihood, reconstruction error).

eric-tramel commented 8 years ago

Okay, all of this got merged, so we're good to go with the validation sets.