diff --git a/tutorials/llm_clinical_trials/scripts/trial_task.py b/tutorials/llm_clinical_trials/scripts/trial_task.py index 704285ca..27933079 100644 --- a/tutorials/llm_clinical_trials/scripts/trial_task.py +++ b/tutorials/llm_clinical_trials/scripts/trial_task.py @@ -15,7 +15,7 @@ Patient group: Number of patients in the group: Treatment drug or substance: -Treatment dose: +Treatment dose: Treatment frequency of administration: Treatment duration: Outcome: @@ -114,9 +114,9 @@ def parse_responses( matcher.add("Patient_Group", [nlp.make_doc(text) for text in patient_groups]) matcher.add("Patient_Number", [nlp.make_doc(text) for text in patient_numbers]) matcher.add("Treatment_Drug", [nlp.make_doc(text) for text in drugs]) - matcher.add("Dose", [nlp.make_doc(text) for text in doses]) - matcher.add("Frequency", [nlp.make_doc(text) for text in frequencies]) - matcher.add("Duration", [nlp.make_doc(text) for text in durations]) + matcher.add("Treatment_Dose", [nlp.make_doc(text) for text in doses]) + matcher.add("Treatment_Frequency", [nlp.make_doc(text) for text in frequencies]) + matcher.add("Treatment_Duration", [nlp.make_doc(text) for text in durations]) matcher.add("Outcome", [nlp.make_doc(text) for text in outcomes]) matches = matcher(doc, as_spans=True)