Predicting lightning with artificial intelligence