code about RBM