RBM train
RBM train refactor
RBM train refactor