Instructions to use aayush7511/cleaned_context_all_sentences with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use aayush7511/cleaned_context_all_sentences with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="aayush7511/cleaned_context_all_sentences")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("aayush7511/cleaned_context_all_sentences") model = AutoModelForSequenceClassification.from_pretrained("aayush7511/cleaned_context_all_sentences", device_map="auto") - Notebooks
- Google Colab
- Kaggle
cleaned_context_all_sentences
This model is a fine-tuned version of google-bert/bert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 0.3971
- Accuracy: 0.847
- Auc: 0.774
- Precision: 0.667
- Recall: 0.296
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.0002
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 10
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Auc | Precision | Recall |
|---|---|---|---|---|---|---|---|
| 0.4744 | 1.0 | 321 | 0.4293 | 0.833 | 0.712 | 0.7 | 0.122 |
| 0.4415 | 2.0 | 642 | 0.4169 | 0.824 | 0.745 | 0.533 | 0.139 |
| 0.4278 | 3.0 | 963 | 0.4276 | 0.833 | 0.745 | 0.722 | 0.113 |
| 0.4106 | 4.0 | 1284 | 0.4071 | 0.844 | 0.761 | 0.653 | 0.278 |
| 0.4081 | 5.0 | 1605 | 0.4005 | 0.844 | 0.763 | 0.623 | 0.33 |
| 0.4073 | 6.0 | 1926 | 0.3974 | 0.841 | 0.766 | 0.633 | 0.27 |
| 0.3931 | 7.0 | 2247 | 0.3953 | 0.849 | 0.774 | 0.705 | 0.27 |
| 0.3919 | 8.0 | 2568 | 0.3965 | 0.849 | 0.769 | 0.661 | 0.322 |
| 0.3862 | 9.0 | 2889 | 0.4061 | 0.842 | 0.773 | 0.706 | 0.209 |
| 0.3794 | 10.0 | 3210 | 0.3971 | 0.847 | 0.774 | 0.667 | 0.296 |
Framework versions
- Transformers 4.55.4
- Pytorch 2.8.0
- Datasets 4.0.0
- Tokenizers 0.21.4
- Downloads last month
- 2
Model tree for aayush7511/cleaned_context_all_sentences
Base model
google-bert/bert-base-uncased