Optimization of computational budget for power system risk assessment