How to optimize an AI algorithm