Edit model card

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
0
Safetensors
Model size
77M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for kmanaa/t5-dialogue-classification-4

Finetuned
(297)
this model