The dataset used in the paper is the BIOS, which is a comprehensive biomedical knowledge graph including diseases, symptoms, medications, surgeries, and non-surgical treatments.
The BIOS dataset is a biography classification dataset with 28 profession classes and 4 intersectional demographic groups (male-wealthy, male-rest, female-wealthy, female-rest).