AI-based rainfall prediction