Instructions to use jtviegas/gpt2classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use jtviegas/gpt2classifier with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import GPT2Classifier model = GPT2Classifier.from_pretrained("jtviegas/gpt2classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
gpt2classifier
This model is a fine-tuned version of on the None dataset. It achieves the following results on the evaluation set:
- Loss: 1.4843
- Accuracy: 0.6701
- Precision: 0.5613
- Recall: 0.5056
- F: 0.5166
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.00014602046726399224
- train_batch_size: 4
- eval_batch_size: 16
- seed: 53
- 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
- lr_scheduler_warmup_steps: 0.1044952152853707
- num_epochs: 7
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F |
|---|---|---|---|---|---|---|---|
| 0.7805 | 1.0 | 969 | 0.8852 | 0.6289 | 0.3703 | 0.4036 | 0.3788 |
| 0.7105 | 2.0 | 1938 | 0.8182 | 0.6351 | 0.3909 | 0.4471 | 0.4168 |
| 0.8493 | 3.0 | 2907 | 0.8963 | 0.6237 | 0.5174 | 0.5303 | 0.5226 |
| 0.6749 | 4.0 | 3876 | 0.8120 | 0.6567 | 0.4100 | 0.4727 | 0.4376 |
| 0.5408 | 5.0 | 4845 | 1.0932 | 0.6680 | 0.5425 | 0.4911 | 0.5019 |
| 0.3422 | 6.0 | 5814 | 1.3366 | 0.6546 | 0.5292 | 0.4946 | 0.4981 |
| 0.4447 | 7.0 | 6783 | 1.4843 | 0.6701 | 0.5613 | 0.5056 | 0.5166 |
Framework versions
- Transformers 5.12.1
- Pytorch 2.12.0+cu130
- Datasets 5.0.0
- Tokenizers 0.22.2
- Downloads last month
- 9
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support