updated number of classes dataset name as well as the eval logs

README.md

@@ -3,7 +3,7 @@ library_name: transformers
 tags:
 - tone
 datasets:
-- Dc-4nderson/tone_dataset
+- Dc-4nderson/tone_dataset2
 language:
 - en
 metrics:
@@ -16,7 +16,7 @@ pipeline_tag: text-classification
 
 DistilBERT Tone Classification Model 
 
-This model fine-tunes distilbert-base-uncased to classify tone into 8 categories relevant to community and mentorship transcripts.
+This model fine-tunes distilbert-base-uncased to classify tone into 7 categories relevant to community and mentorship transcripts.
 
 📌 Labels
 
@@ -49,7 +49,7 @@ Optimizer: AdamW (lr=2e-5)
 
 Batch size: 16
 
-Epochs: 5
+Epochs: 8
 
 Loss: CrossEntropy
 
@@ -57,25 +57,25 @@ Metrics: Accuracy + Weighted F1
 
 📈 Validation Metrics
 Epoch	Training Loss	Validation Loss  Accuracy	F1
-1	    No log	          1.260710	     0.801242	  0.784157
-2	    No log	          0.777540	     0.869565	  0.869093
-3	    No log	          0.577972	     0.869565	  0.868584
-4	    No log	          0.481008	     0.900621	  0.900356
-5	    No log	          0.452635	     0.900621	  0.900356
-
+1	          No log	   1.281651	    0.782288	0.778880
+2	          No log	   0.779447	    0.845018	0.843397
+3	          No log	   0.566092	    0.859779	0.856186
+4	          No log	   0.415437	    0.892989	0.892445
+5	          No log	   0.340598	    0.915129	0.914765
+6             0.729500     0.307513	    0.922509	0.922262
+7	          0.729500	   0.296827	    0.915129	0.915210
+8	          0.729500	   0.285301	    0.922509    0.922262
 
 Final Training Summary:
 
-TrainOutput(
-global_step=205, 
-training_loss=0.8436699843988186,
-metrics={'train_runtime': 17.74,
-'train_samples_per_second': 181.229, 
-'train_steps_per_second': 11.556, 
-'total_flos': 106480165436160.0, 
-'train_loss': 0.8436699843988186, 
-'epoch': 5.0
-})
+TrainOutput(global_step=704,
+training_loss=0.5666945034807379,
+metrics={'train_runtime': 42.6317,
+'train_samples_per_second': 261.402,
+'train_steps_per_second': 16.514,
+'total_flos': 369087080441856.0, 
+'train_loss': 0.5666945034807379, 
+'epoch': 8.0})
 
 💻 Usage
 from transformers import pipeline