Added sparse_amax to main bag_of_words