added rbm model training