t5-dialogue-classification-4
This model is a fine-tuned version of google/flan-t5-small on an unknown dataset. It achieves the following results on the evaluation set:
Loss: 0.6143
Classification Report: precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4 ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.5000 0.2500 0.3333 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8077 0.9545 0.8750 22 GENHX 0.5429 0.9500 0.6909 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6667 0.8571 0.7500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.5714 1.0000 0.7273 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6400 100
macro avg 0.3164 0.3187 0.2986 100
weighted avg 0.5702 0.6400 0.5690 100
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0001
- train_batch_size: 7
- eval_batch_size: 7
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 20
Training results
| Training Loss | Epoch | Step | Validation Loss | Classification Report |
|---|---|---|---|---|
| No log | 1.0 | 172 | 0.7682 | precision recall f1-score support |
ALLERGY 0.3333 0.5000 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4 Afib 0.0000 0.0000 0.0000 0 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 0.0000 0.0000 0.0000 1 FAM/SOCHX 0.7500 0.6818 0.7143 22 GENHX 0.3636 1.0000 0.5333 20 GYNHX 0.0000 0.0000 0.0000 1 ICT 0.0000 0.0000 0.0000 0 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 Iv 0.0000 0.0000 0.0000 0 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 1.0000 0.1429 0.2500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.1250 0.2500 0.1667 4 PASTSURGICAL 0.5714 0.5000 0.5333 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.0000 0.0000 0.0000 11
accuracy 0.4300 100
macro avg 0.1367 0.1337 0.1129 100
weighted avg 0.3718 0.4300 0.3466 100 | | No log | 2.0 | 344 | 0.5464 | precision recall f1-score support
ALLERGY 0.5000 0.2500 0.3333 4
ASSESSMENT 0.0000 0.0000 0.0000 4
Afib 0.0000 0.0000 0.0000 0
CC 0.0000 0.0000 0.0000 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 0.0000 0.0000 0.0000 1
FAM/SOCHX 0.8750 0.9545 0.9130 22
GENHX 0.4524 0.9500 0.6129 20
GYNHX 0.0000 0.0000 0.0000 1
I am going home with my son 0.0000 0.0000 0.0000 0 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.4545 0.7143 0.5556 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.3750 0.7500 0.5000 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.0000 0.0000 0.0000 11 thoughts 0.0000 0.0000 0.0000 0
accuracy 0.5700 100
macro avg 0.1542 0.2008 0.1677 100
weighted avg 0.4209 0.5700 0.4710 100
|
| 0.5961 | 3.0 | 516 | 0.4964 | precision recall f1-score support
ALLERGY 1.0000 0.2500 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 0.0000 0.0000 0.0000 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 0.0000 0.0000 0.0000 1
FAM/SOCHX 0.8750 0.9545 0.9130 22
GENHX 0.4878 1.0000 0.6557 20
GYNHX 0.0000 0.0000 0.0000 1
How are you feeling? 0.0000 0.0000 0.0000 0
I V 0.0000 0.0000 0.0000 0
I am going home with my son. 0.0000 0.0000 0.0000 0 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.7143 0.7143 0.7143 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.4444 1.0000 0.6154 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.0000 0.0000 0.0000 11
accuracy 0.5900 100
macro avg 0.1918 0.2139 0.1843 100
weighted avg 0.4689 0.5900 0.4979 100
| | 0.5961 | 4.0 | 688 | 0.5069 | precision recall f1-score support
ALLERGY 1.0000 0.2500 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4 Afib 0.0000 0.0000 0.0000 0 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8462 1.0000 0.9167 22 GENHX 0.6333 0.9500 0.7600 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6667 0.8571 0.7500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.4000 1.0000 0.5714 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 1.0000 0.1818 0.3077 11
accuracy 0.6300 100
macro avg 0.3064 0.2971 0.2689 100
weighted avg 0.6066 0.6300 0.5642 100 | | 0.5961 | 5.0 | 860 | 0.4889 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4 Afib 0.0000 0.0000 0.0000 0 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8462 1.0000 0.9167 22 GENHX 0.6129 0.9500 0.7451 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.8333 0.7143 0.7692 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.3000 0.7500 0.4286 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11
accuracy 0.6200 100
macro avg 0.2928 0.2903 0.2740 100
weighted avg 0.5735 0.6200 0.5651 100 | | 0.1563 | 6.0 | 1032 | 0.4923 | precision recall f1-score support
ALLERGY 1.0000 0.2500 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8462 1.0000 0.9167 22 GENHX 0.5128 1.0000 0.6780 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.5556 0.7143 0.6250 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.3750 0.7500 0.5000 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 1.0000 0.1818 0.3077 11
accuracy 0.6200 100
macro avg 0.3089 0.2948 0.2684 100
weighted avg 0.5737 0.6200 0.5362 100 | | 0.1563 | 7.0 | 1204 | 0.5146 | precision recall f1-score support
ALLERGY 1.0000 0.2500 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8750 0.9545 0.9130 22 GENHX 0.5000 1.0000 0.6667 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 Injection 0.0000 0.0000 0.0000 0 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6250 0.7143 0.6667 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.5000 0.7500 0.6000 4 PASTSURGICAL 1.0000 1.0000 1.0000 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11
accuracy 0.6100 100
macro avg 0.2937 0.2786 0.2634 100
weighted avg 0.5596 0.6100 0.5423 100 | | 0.1563 | 8.0 | 1376 | 0.5325 | precision recall f1-score support
ALLERGY 1.0000 0.2500 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8000 0.9091 0.8511 22 GENHX 0.5128 1.0000 0.6780 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6000 0.8571 0.7059 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.6000 0.7500 0.6667 4 PASTSURGICAL 1.0000 1.0000 1.0000 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11
accuracy 0.6100 100
macro avg 0.3090 0.2974 0.2794 100
weighted avg 0.5479 0.6100 0.5363 100 | | 0.0973 | 9.0 | 1548 | 0.5105 | precision recall f1-score support
ALLERGY 1.0000 0.2500 0.4000 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8400 0.9545 0.8936 22 GENHX 0.5429 0.9500 0.6909 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6000 0.8571 0.7059 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.6000 0.7500 0.6667 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11 harming 0.0000 0.0000 0.0000 0
accuracy 0.6100 100
macro avg 0.2923 0.2830 0.2659 100
weighted avg 0.5538 0.6100 0.5436 100 | | 0.0973 | 10.0 | 1720 | 0.5433 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8148 1.0000 0.8980 22 GENHX 0.6129 0.9500 0.7451 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6667 0.8571 0.7500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.5000 1.0000 0.6667 4 PASTSURGICAL 1.0000 1.0000 1.0000 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7500 0.2727 0.4000 11 harming 0.0000 0.0000 0.0000 0
accuracy 0.6500 100
macro avg 0.3021 0.3133 0.2917 100
weighted avg 0.5810 0.6500 0.5864 100 | | 0.0973 | 11.0 | 1892 | 0.5202 | precision recall f1-score support
ALLERGY 0.6667 0.5000 0.5714 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.0000 0.0000 0.0000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.8148 1.0000 0.8980 22 GENHX 0.5714 1.0000 0.7273 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6250 0.7143 0.6667 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.5714 1.0000 0.7273 4 PASTSURGICAL 0.8889 1.0000 0.9412 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11 harming 0.0000 0.0000 0.0000 0
accuracy 0.6400 100
macro avg 0.2764 0.3046 0.2770 100
weighted avg 0.5413 0.6400 0.5583 100 | | 0.0612 | 12.0 | 2064 | 0.5594 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.5000 0.2500 0.3333 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.7857 1.0000 0.8800 22 GENHX 0.5882 1.0000 0.7407 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6667 0.8571 0.7500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.7500 0.7500 0.7500 4 PASTSURGICAL 0.6667 1.0000 0.8000 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11 harming 0.0000 0.0000 0.0000 0
accuracy 0.6500 100
macro avg 0.3154 0.3114 0.2955 100
weighted avg 0.5638 0.6500 0.5697 100 | | 0.0612 | 13.0 | 2236 | 0.5590 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 1.0000 0.2500 0.4000 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.7778 0.9545 0.8571 22 GENHX 0.5714 1.0000 0.7273 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6667 0.8571 0.7500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.5714 1.0000 0.7273 4 PASTSURGICAL 0.8000 1.0000 0.8889 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11 harming 0.0000 0.0000 0.0000 0
accuracy 0.6500 100
macro avg 0.3359 0.3211 0.3001 100
weighted avg 0.5823 0.6500 0.5708 100 | | 0.0612 | 14.0 | 2408 | 0.5971 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4 CC 0.5000 0.2500 0.3333 4 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 2 EDCOURSE 0.0000 0.0000 0.0000 3 EXAM 1.0000 1.0000 1.0000 1 FAM/SOCHX 0.7857 1.0000 0.8800 22 GENHX 0.5882 1.0000 0.7407 20 GYNHX 0.0000 0.0000 0.0000 1 IMAGING 0.0000 0.0000 0.0000 1 IMMUNIZATIONS 0.0000 0.0000 0.0000 1 LABS 0.0000 0.0000 0.0000 1 MEDICATIONS 0.6667 0.8571 0.7500 7 OTHER_HISTORY 0.0000 0.0000 0.0000 1 PASTMEDICALHX 0.4286 0.7500 0.5455 4 PASTSURGICAL 1.0000 1.0000 1.0000 8 PLAN 0.0000 0.0000 0.0000 3 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6667 0.1818 0.2857 11 harming 0.0000 0.0000 0.0000 0
accuracy 0.6500 100
macro avg 0.3160 0.3114 0.2953 100
weighted avg 0.5776 0.6500 0.5775 100 | | 0.0442 | 15.0 | 2580 | 0.6045 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 0.0000 0.0000 0.0000 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 1.0000 1.0000 1.0000 1
FAM/SOCHX 0.8077 0.9545 0.8750 22
GENHX 0.5263 1.0000 0.6897 20
GYNHX 0.0000 0.0000 0.0000 1
IMAGING 0.0000 0.0000 0.0000 1
IMMUNIZATIONS 0.0000 0.0000 0.0000 1
LABS 0.0000 0.0000 0.0000 1
MEDICATIONS 0.6667 0.8571 0.7500 7
OTHER_HISTORY 0.0000 0.0000 0.0000 1
PASTMEDICALHX 0.5000 0.7500 0.6000 4
PASTSURGICAL 1.0000 1.0000 1.0000 8
PLAN 0.0000 0.0000 0.0000 3
PROCEDURES 0.0000 0.0000 0.0000 1
ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6300 100
macro avg 0.2937 0.2973 0.2794 100
weighted avg 0.5530 0.6300 0.5550 100
|
| 0.0442 | 16.0 | 2752 | 0.6341 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 0.0000 0.0000 0.0000 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 1.0000 1.0000 1.0000 1
FAM/SOCHX 0.7778 0.9545 0.8571 22
GENHX 0.5405 1.0000 0.7018 20
GYNHX 0.0000 0.0000 0.0000 1
IMAGING 0.0000 0.0000 0.0000 1
IMMUNIZATIONS 0.0000 0.0000 0.0000 1
LABS 0.0000 0.0000 0.0000 1
MEDICATIONS 0.6667 0.8571 0.7500 7
OTHER_HISTORY 0.0000 0.0000 0.0000 1
PASTMEDICALHX 0.5000 0.7500 0.6000 4
PASTSURGICAL 1.0000 1.0000 1.0000 8
PLAN 0.0000 0.0000 0.0000 3
PROCEDURES 0.0000 0.0000 0.0000 1
ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6300 100
macro avg 0.2929 0.2973 0.2791 100
weighted avg 0.5492 0.6300 0.5535 100
|
| 0.0442 | 17.0 | 2924 | 0.6230 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 0.3333 0.2500 0.2857 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 1.0000 1.0000 1.0000 1
FAM/SOCHX 0.8077 0.9545 0.8750 22
GENHX 0.5429 0.9500 0.6909 20
GYNHX 0.0000 0.0000 0.0000 1
IMAGING 0.0000 0.0000 0.0000 1
IMMUNIZATIONS 0.0000 0.0000 0.0000 1
LABS 0.0000 0.0000 0.0000 1
MEDICATIONS 0.6667 0.8571 0.7500 7
OTHER_HISTORY 0.0000 0.0000 0.0000 1
PASTMEDICALHX 0.5000 0.7500 0.6000 4
PASTSURGICAL 1.0000 1.0000 1.0000 8
PLAN 0.0000 0.0000 0.0000 3
PROCEDURES 0.0000 0.0000 0.0000 1
ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6300 100
macro avg 0.3103 0.3068 0.2930 100
weighted avg 0.5696 0.6300 0.5667 100
|
| 0.0266 | 18.0 | 3096 | 0.6051 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 1.0000 0.2500 0.4000 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 1.0000 1.0000 1.0000 1
FAM/SOCHX 0.7778 0.9545 0.8571 22
GENHX 0.5556 1.0000 0.7143 20
GYNHX 0.0000 0.0000 0.0000 1
IMAGING 0.0000 0.0000 0.0000 1
IMMUNIZATIONS 0.0000 0.0000 0.0000 1
LABS 0.0000 0.0000 0.0000 1
MEDICATIONS 0.6667 0.8571 0.7500 7
OTHER_HISTORY 0.0000 0.0000 0.0000 1
PASTMEDICALHX 0.5714 1.0000 0.7273 4
PASTSURGICAL 0.8889 1.0000 0.9412 8
PLAN 0.0000 0.0000 0.0000 3
PROCEDURES 0.0000 0.0000 0.0000 1
ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6500 100
macro avg 0.3394 0.3211 0.3020 100
weighted avg 0.5862 0.6500 0.5724 100
|
| 0.0266 | 19.0 | 3268 | 0.6057 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 1.0000 0.2500 0.4000 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 1.0000 1.0000 1.0000 1
FAM/SOCHX 0.7778 0.9545 0.8571 22
GENHX 0.5556 1.0000 0.7143 20
GYNHX 0.0000 0.0000 0.0000 1
IMAGING 0.0000 0.0000 0.0000 1
IMMUNIZATIONS 0.0000 0.0000 0.0000 1
LABS 0.0000 0.0000 0.0000 1
MEDICATIONS 0.6667 0.8571 0.7500 7
OTHER_HISTORY 0.0000 0.0000 0.0000 1
PASTMEDICALHX 0.5714 1.0000 0.7273 4
PASTSURGICAL 0.8889 1.0000 0.9412 8
PLAN 0.0000 0.0000 0.0000 3
PROCEDURES 0.0000 0.0000 0.0000 1
ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6500 100
macro avg 0.3394 0.3211 0.3020 100
weighted avg 0.5862 0.6500 0.5724 100
|
| 0.0266 | 20.0 | 3440 | 0.6143 | precision recall f1-score support
ALLERGY 1.0000 0.5000 0.6667 4
ASSESSMENT 0.0000 0.0000 0.0000 4
CC 0.5000 0.2500 0.3333 4
DIAGNOSIS 0.0000 0.0000 0.0000 1
DISPOSITION 0.0000 0.0000 0.0000 2
EDCOURSE 0.0000 0.0000 0.0000 3
EXAM 1.0000 1.0000 1.0000 1
FAM/SOCHX 0.8077 0.9545 0.8750 22
GENHX 0.5429 0.9500 0.6909 20
GYNHX 0.0000 0.0000 0.0000 1
IMAGING 0.0000 0.0000 0.0000 1
IMMUNIZATIONS 0.0000 0.0000 0.0000 1
LABS 0.0000 0.0000 0.0000 1
MEDICATIONS 0.6667 0.8571 0.7500 7
OTHER_HISTORY 0.0000 0.0000 0.0000 1
PASTMEDICALHX 0.5714 1.0000 0.7273 4
PASTSURGICAL 0.8889 1.0000 0.9412 8
PLAN 0.0000 0.0000 0.0000 3
PROCEDURES 0.0000 0.0000 0.0000 1
ROS 0.6667 0.1818 0.2857 11
harming yourself or others 0.0000 0.0000 0.0000 0
accuracy 0.6400 100
macro avg 0.3164 0.3187 0.2986 100
weighted avg 0.5702 0.6400 0.5690 100
|
Framework versions
- Transformers 4.38.2
- Pytorch 2.1.0+cu121
- Datasets 2.18.0
- Tokenizers 0.15.2
- Downloads last month
- 9
Model tree for kmanaa/t5-dialogue-classification-4
Base model
google/flan-t5-small