Model save

README.md

@@ -14,7 +14,7 @@ model-index:
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should proofread and complete it, then remove this comment. -->
+should probably proofread and complete it, then remove this comment. -->
 
 # Mistral-7B-v0.1_Emotion
 
@@ -22,11 +22,7 @@ This model is a fine-tuned version of [mistralai/Mistral-7B-Instruct-v0.1](https
 
 ## Model description
 
-https://ai.plainenglish.io/fine-tuning-the-mistral-7b-instruct-v0-1-model-with-the-emotion-dataset-c84c50b553dc
-
-Fine tunning: https://github.com/frank-morales2020/MLxDL/blob/main/FineTuning_Mistral_7b_hfdeployment_dataset_Emotion.ipynb
-
-Evaluation: https://github.com/frank-morales2020/MLxDL/blob/main/FineTunning_Testing_For_EmotionQADataset.ipynb
+More information needed
 
 ## Intended uses & limitations
 
@@ -34,45 +30,7 @@ More information needed
 
 ## Training and evaluation data
 
-https://github.com/frank-morales2020/MLxDL/blob/main/FineTunning_Testing_For_EmotionQADataset.ipynb
-
-The following hyperparameters were used during training:
-
------------
-
-learning_rate: 0.0002 train_batch_size: 3 eval_batch_size: 8 seed: 42 gradient_accumulation_steps: 2 
-total_train_batch_size: 6 optimizer: Adam with betas=(0.9,0.999) 
-and epsilon=1e-08 lr_scheduler_type: constant lr_scheduler_warmup_ratio: 0.03 
-
-num_epochs: 1
-
-NOTE: test - Accuracy (Eval dataset and predict) for a sample of 2000: 59.45%
-
------------
-
-The following hyperparameters were used during training:
-
-learning_rate: 0.0002 train_batch_size: 3 eval_batch_size: 8 seed: 42 gradient_accumulation_steps: 2 
-total_train_batch_size: 6 optimizer: Adam with betas=(0.9,0.999) 
-and epsilon=1e-08 lr_scheduler_type: constant lr_scheduler_warmup_ratio: 0.03 
-
-num_epochs: 25
-
-NOTE: test - Accuracy (Eval dataset and predict) for a sample of 2000: 79.95%
-
------------
-
-The following hyperparameters were used during training:
-
-learning_rate: 0.0002 train_batch_size: 3 eval_batch_size: 8 seed: 42 gradient_accumulation_steps: 2 
-total_train_batch_size: 6 optimizer: Adam with betas=(0.9,0.999) 
-and epsilon=1e-08 lr_scheduler_type: constant lr_scheduler_warmup_ratio: 0.03 
-
-num_epochs: 100
-
-NOTE: test - Accuracy (Eval dataset and predict) for a sample of 2000: TBD
-
------------
+More information needed
 
 ## Training procedure
 
@@ -88,7 +46,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: constant
 - lr_scheduler_warmup_ratio: 0.03
-- num_epochs: 25
+- num_epochs: 40
 
 ### Training results