Assessment of machine learning methods for state-to-state approaches