Update README.md

README.md

@@ -8,12 +8,13 @@ tags:
 datasets:
 - disham993/ElectricalDeviceFeedbackBalanced
 metrics:
-- epoch: 5.0
+- epoch: 5
 - eval_f1: 0.9106
 - eval_accuracy: 0.90828
 - eval_runtime: 3.2658
 - eval_samples_per_second: 413.989
 - eval_steps_per_second: 13.167
+library_name: transformers
 ---
 
 # electrical-classification-ModernBERT-large
@@ -24,7 +25,7 @@ This model is fine-tuned from [answerdotai/ModernBERT-large](https://huggingface
 
 ## Training Data
 
-The model was trained on the [disham993/ElectricalDeviceFeedbackBalanced](https://huggingface.co/datasets/disham993/ElectricalDeviceFeedbackBalanced) dataset, which has been carefully balanced to address class imbalances effectively.
+The model was trained on the [disham993/ElectricalDeviceFeedbackBalanced](https://huggingface.co/datasets/disham993/ElectricalDeviceFeedbackBalanced) dataset, which has been carefully balanced to address class imbalances effectively. Original dataset which is imbalanced: [disham993/ElectricalDeviceFeedback](https://huggingface.co/datasets/disham993/ElectricalDeviceFeedback).
 
 ## Model Details
 - **Base Model:** [answerdotai/ModernBERT-large](https://huggingface.co/answerdotai/ModernBERT-large) 
@@ -84,4 +85,4 @@ For a complete guide covering the entire process - from data tokenization to pus
 
 ## Last update
 
-2025-01-05
+2025-01-05