o 7t¾iE ã@spddlZddlZdejfdd„Zdejfdd„Zdd„Zd efd d „Zd d „Zd efdd„Z dd„Z dd„Z dS)éNÚdatasetcCó|S©N©)rrrú_/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/tasks/med_text_classification/utils.pyÚprocess_docs_hardórcCs4dd„}t|ƒ}td|ƒ}| t|ƒ¡}| |¡S)NcSrrr©ÚdocrrrÚ_helper rzprocess_docs.._helpergš™™™™™¹?)ÚlenÚintÚselectÚrangeÚmap)rr Ú num_entriesÚten_percent_indexÚfiltered_datasetrrrÚ process_docs s   rcCsgd¢S)N)Ú neoplasmszdigestive system diseasesznervous system diseaseszcardiovascular diseaseszgeneral pathological conditionsrr rrrÚdoc_to_choice_easysrÚreturnc Cs8t|ƒ}d |d|d|d|d|d|d¡}|S)Nz”Classify the topic of the following medical text into one of the following choices. Text: {} Choices: A. {} B. {} C. {} D. {} E. {} Answer:Útextréééé)rÚformat©r ÚchoicesÚpromptrrrÚdoc_to_text_easy!s$ÿø r!cCst|dƒdS)NÚclassr)r r rrrÚdoc_to_target_easy3sr#cCst|ƒ}d |d|¡}|S)Nz}Select the medical specialty the following text is talking about among the following choices. Text: {} Choices: {} Answer:Ú transcription)Údoc_to_choice_hardrrrrrÚdoc_to_text_hard7s  ür&cCs gd¢}|S)N)(z Bariatricsz Allergy / Immunologyz Dentistryz Cardiovascular / Pulmonaryz Urologyz Hospice - Palliative Carez Radiologyz Pediatrics - Neonatalz Neurologyz Neurosurgeryz Emergency Room Reportsz IME-QME-Work Comp etc.z Office Notesz Surgeryz Lettersz Ophthalmologyz Hematology - Oncologyz Endocrinologyz Cosmetic / Plastic Surgeryz Diets and Nutritionsz Rheumatologyz Nephrologyz Physical Medicine - Rehabz Podiatryz Chiropracticz Lab Medicine - Pathologyz Orthopedicz Autopsyz Psychiatry / Psychologyz Speech - Languagez ENT - Otolaryngologyz Sleep Medicinez Dermatologyz SOAP / Chart / Progress Notesz General Medicinez Consult - History and Phy.z Obstetrics / Gynecologyz Gastroenterologyz Pain Managementz Discharge Summaryr)r Ú choices_listrrrr%Cs*r%cCst|ƒ}|d}| |¡}|S)NÚmedical_specialty)r%Úindex)r rÚgoldÚidxrrrÚdoc_to_target_hardqs r,) ÚrandomÚdatasetsÚDatasetrrrÚstrr!r#r&r%r,rrrrÚs   .