Stratified-NMF for Heterogeneous Data