Machine learning applied to massive stars