Combining ab initio and machine learning method to improve prediction performance of diatomic vibrational energies