Artificial intelligence trained to find disease-related genes