We divide input features into four categories: PSA, age, race, and cancer stages. Here, KLK3 is linked to cancer.