Machine learning-based Raman amplifier design