This project aims to model different intervention policies during COVID-19 and provide a policy-based recommendation on optimizing governments’ vaccination campaigns. By taking the example of India’s COVID-19 scenarios, we show that vaccination is the most effective intervention policy. Moreover, we provide a machine learning-based policy recommendation method on the vaccination campaign of COVID-19 by minimizing three different cost factors: the duration of the pandemic, the budget of the COVID-19 battle as well as the death toll. We validate our method based on the real-world dataset of India by comparing our simulated results with the government’s vaccination plan from machine learning prediction. Our approach shows a 13% decrease in disease control time and government budget. At the same time, we find out that vaccination based on each province’s population leads to a 12.4% decrease in the death toll than on infection cases. The model developed in this study has practical implications for COVID-19 vaccination campaigns and the infection control of other infectious diseases.