The atom-pair matrix of Coulomb interaction was used as input. The training was carried out with 5000 molecules, and the number of the hidden layers of the model was four. The RMSE values of their model were 0.12 and 1.76 eV for the maximum oscillator strength and corresponding excitation energy, respectively. The accuracy of our model outperforms their model in terms of both oscillator strength and excitation energy prediction. Notably, the RMSE of our model in excitation energy prediction is only one-third of their model.