metadata

license: apache-2.0
base_model: google/flan-t5-small
tags:
  - generated_from_trainer
model-index:
  - name: t5-dialogue-classification-enriched
    results: []

t5-dialogue-classification-enriched

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.3777
Classification Report: precision recall f1-score support
```
ALLERGY     0.9167    0.9167    0.9167        12
```
ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.5000 0.3636 0.4211 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.8246 0.8868 0.8545 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5652 0.9286 0.7027 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7550       200
macro avg     0.3570    0.3589    0.3519       200

weighted avg 0.7153 0.7550 0.7277 200

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.3252	precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.0000 0.0000 0.0000 11 CC 0.2727 0.2727 0.2727 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.0000 0.0000 0.0000 5 FAM/SOCHX 0.8571 0.9333 0.8936 45 GENHX 0.6923 0.8491 0.7627 53 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.8182 0.9000 0.8571 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.3158 0.8571 0.4615 14 PASTSURGICAL 0.8750 1.0000 0.9333 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.4000 0.1176 0.1818 17

 accuracy                         0.6550       200
macro avg     0.2616    0.2923    0.2660       200

weighted avg 0.5790 0.6550 0.5989 200 | | No log | 2.0 | 344 | 0.2772 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.0000 0.0000 0.0000 11 CC 0.4000 0.5455 0.4615 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.0000 0.0000 0.0000 5 FAM/SOCHX 0.8269 0.9556 0.8866 45 GENHX 0.7541 0.8679 0.8070 53 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 1.0000 0.9091 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.4615 0.8571 0.6000 14 PASTSURGICAL 1.0000 1.0000 1.0000 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.6923 0.5294 0.6000 17

 accuracy                         0.7200       200
macro avg     0.2984    0.3336    0.3110       200

weighted avg 0.6357 0.7200 0.6696 200 | | 0.2496 | 3.0 | 516 | 0.2897 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.0000 0.0000 0.0000 11 CC 0.3333 0.3636 0.3478 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.5000 0.2000 0.2857 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.6216 0.8679 0.7244 53 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 1.0000 0.9091 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.6250 0.7143 0.6667 14 PASTSURGICAL 0.7000 1.0000 0.8235 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7500 0.5294 0.6207 17

 accuracy                         0.7100       200
macro avg     0.3113    0.3285    0.3126       200

weighted avg 0.6233 0.7100 0.6556 200 | | 0.2496 | 4.0 | 688 | 0.2703 | precision recall f1-score support

  ALLERGY     0.9167    0.9167    0.9167        12

ASSESSMENT 0.0000 0.0000 0.0000 11 CC 0.5556 0.4545 0.5000 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.5000 0.4000 0.4444 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.7344 0.8868 0.8034 53 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 1.0000 0.9091 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5000 0.8571 0.6316 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7647 0.7647 0.7647 17

 accuracy                         0.7450       200
macro avg     0.3332    0.3546    0.3399       200

weighted avg 0.6628 0.7450 0.6972 200 | | 0.2496 | 5.0 | 860 | 0.2670 | precision recall f1-score support

  ALLERGY     0.9167    0.9167    0.9167        12

ASSESSMENT 0.0000 0.0000 0.0000 11 CC 0.4545 0.4545 0.4545 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 1.0000 0.2500 0.4000 4 EXAM 0.5000 0.4000 0.4444 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.8036 0.8491 0.8257 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5263 0.7143 0.6061 14 PASTSURGICAL 0.5385 1.0000 0.7000 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7647 0.7647 0.7647 17

 accuracy                         0.7350       200
macro avg     0.3567    0.3652    0.3443       200

weighted avg 0.6781 0.7350 0.6970 200 | | 0.1353 | 6.0 | 1032 | 0.2786 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.0000 0.0000 0.0000 11 CC 0.6250 0.4545 0.5263 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 1.0000 0.5000 0.6667 4 EXAM 0.2500 0.2000 0.2222 5 FAM/SOCHX 0.8776 0.9556 0.9149 45 GENHX 0.7966 0.8868 0.8393 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.4583 0.7857 0.5789 14 PASTSURGICAL 0.6364 1.0000 0.7778 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7647 0.7647 0.7647 17

 accuracy                         0.7500       200
macro avg     0.3589    0.3732    0.3558       200

weighted avg 0.6870 0.7500 0.7097 200 | | 0.1353 | 7.0 | 1204 | 0.2730 | precision recall f1-score support

  ALLERGY     0.9167    0.9167    0.9167        12

ASSESSMENT 0.5000 0.0909 0.1538 11 CC 0.5000 0.5455 0.5217 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.1667 0.2000 0.1818 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.7797 0.8679 0.8214 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5455 0.8571 0.6667 14 PASTSURGICAL 0.8571 0.8571 0.8571 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7400       200
macro avg     0.3347    0.3498    0.3322       200

weighted avg 0.6889 0.7400 0.7020 200 | | 0.1353 | 8.0 | 1376 | 0.3150 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5000 0.0909 0.1538 11 CC 0.4167 0.4545 0.4348 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.1667 0.2000 0.1818 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.8036 0.8491 0.8257 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5000 0.9286 0.6500 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7857 0.6471 0.7097 17

 accuracy                         0.7350       200
macro avg     0.3402    0.3461    0.3311       200

weighted avg 0.6969 0.7350 0.7010 200 | | 0.0858 | 9.0 | 1548 | 0.3058 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.6667 0.1818 0.2857 11 CC 0.7500 0.2727 0.4000 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.3333 1.0000 0.5000 1 EDCOURSE 1.0000 0.2500 0.4000 4 EXAM 0.2500 0.2000 0.2222 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.7541 0.8679 0.8070 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5200 0.9286 0.6667 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7500       200
macro avg     0.4353    0.4079    0.3849       200

weighted avg 0.7376 0.7500 0.7175 200 | | 0.0858 | 10.0 | 1720 | 0.2911 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.4286 0.2727 0.3333 11 CC 0.4444 0.3636 0.4000 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.5000 0.2500 0.3333 4 EXAM 0.5000 0.2000 0.2857 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.7759 0.8491 0.8108 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5455 0.8571 0.6667 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7450       200
macro avg     0.3813    0.3625    0.3623       200

weighted avg 0.7099 0.7450 0.7189 200 | | 0.0858 | 11.0 | 1892 | 0.3449 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5000 0.2727 0.3529 11 CC 0.8000 0.3636 0.5000 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.3333 1.0000 0.5000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.2500 0.2000 0.2222 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.7833 0.8868 0.8319 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5000 0.8571 0.6316 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7857 0.6471 0.7097 17

 accuracy                         0.7500       200
macro avg     0.3792    0.3989    0.3707       200

weighted avg 0.7164 0.7500 0.7194 200 | | 0.0533 | 12.0 | 2064 | 0.3194 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.7143 0.4545 0.5556 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.8070 0.8679 0.8364 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5217 0.8571 0.6486 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8125 0.7647 0.7879 17

 accuracy                         0.7600       200
macro avg     0.3696    0.3630    0.3594       200

weighted avg 0.7261 0.7600 0.7347 200 | | 0.0533 | 13.0 | 2236 | 0.3398 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.6000 0.2727 0.3750 11 CC 0.5000 0.3636 0.4211 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.5000 0.2500 0.3333 4 EXAM 1.0000 0.2000 0.3333 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.7460 0.8868 0.8103 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5714 0.8571 0.6857 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7500       200
macro avg     0.4173    0.3633    0.3682       200

weighted avg 0.7282 0.7500 0.7221 200 | | 0.0533 | 14.0 | 2408 | 0.3339 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.6250 0.4545 0.5263 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.8246 0.8868 0.8545 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.6500 0.9286 0.7647 14 PASTSURGICAL 0.7778 1.0000 0.8750 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7700       200
macro avg     0.3607    0.3717    0.3604       200

weighted avg 0.7259 0.7700 0.7411 200 | | 0.0367 | 15.0 | 2580 | 0.3354 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.4000 0.1818 0.2500 11 CC 0.5000 0.5455 0.5217 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.5000 0.2500 0.3333 4 EXAM 0.5000 0.2000 0.2857 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.8364 0.8679 0.8519 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.6500 0.9286 0.7647 14 PASTSURGICAL 0.8750 1.0000 0.9333 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.7647 0.7647 0.7647 17

 accuracy                         0.7700       200
macro avg     0.3829    0.3816    0.3724       200

weighted avg 0.7273 0.7700 0.7403 200 | | 0.0367 | 16.0 | 2752 | 0.3511 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.7143 0.4545 0.5556 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.5000 0.2500 0.3333 4 EXAM 0.5000 0.2000 0.2857 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.8246 0.8868 0.8545 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5652 0.9286 0.7027 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8125 0.7647 0.7879 17

 accuracy                         0.7750       200
macro avg     0.4060    0.3800    0.3815       200

weighted avg 0.7479 0.7750 0.7509 200 | | 0.0367 | 17.0 | 2924 | 0.3552 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.6667 0.3636 0.4706 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.8246 0.8868 0.8545 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5652 0.9286 0.7027 14 PASTSURGICAL 0.8750 1.0000 0.9333 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7600       200
macro avg     0.3633    0.3660    0.3568       200

weighted avg 0.7251 0.7600 0.7332 200 | | 0.0258 | 18.0 | 3096 | 0.3670 | precision recall f1-score support

  ALLERGY     1.0000    0.9167    0.9565        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.5000 0.4545 0.4762 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8627 0.9778 0.9167 45 GENHX 0.8333 0.8491 0.8411 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5652 0.9286 0.7027 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7550       200
macro avg     0.3618    0.3627    0.3565       200

weighted avg 0.7232 0.7550 0.7321 200 | | 0.0258 | 19.0 | 3268 | 0.3751 | precision recall f1-score support

  ALLERGY     0.9167    0.9167    0.9167        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.6667 0.3636 0.4706 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.8246 0.8868 0.8545 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5417 0.9286 0.6842 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7550       200
macro avg     0.3642    0.3589    0.3534       200

weighted avg 0.7228 0.7550 0.7291 200 | | 0.0258 | 20.0 | 3440 | 0.3777 | precision recall f1-score support

  ALLERGY     0.9167    0.9167    0.9167        12

ASSESSMENT 0.5714 0.3636 0.4444 11 CC 0.5000 0.3636 0.4211 11 DIAGNOSIS 0.0000 0.0000 0.0000 1 DISPOSITION 0.0000 0.0000 0.0000 1 EDCOURSE 0.0000 0.0000 0.0000 4 EXAM 0.3333 0.2000 0.2500 5 FAM/SOCHX 0.8600 0.9556 0.9053 45 GENHX 0.8246 0.8868 0.8545 53 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.7692 1.0000 0.8696 10 OTHER_HISTORY 0.0000 0.0000 0.0000 3 PASTMEDICALHX 0.5652 0.9286 0.7027 14 PASTSURGICAL 1.0000 0.8571 0.9231 7 PLAN 0.0000 0.0000 0.0000 1 PROCEDURES 0.0000 0.0000 0.0000 1 ROS 0.8000 0.7059 0.7500 17

 accuracy                         0.7550       200
macro avg     0.3570    0.3589    0.3519       200

weighted avg 0.7153 0.7550 0.7277 200 |

Framework versions

Transformers 4.35.2
Pytorch 2.1.0+cu121
Datasets 2.16.1
Tokenizers 0.15.0