Instructions to use Kicel/sparse_imdb_classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use Kicel/sparse_imdb_classifier with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("FacebookAI/xlm-roberta-base") model = PeftModel.from_pretrained(base_model, "Kicel/sparse_imdb_classifier") - Transformers
How to use Kicel/sparse_imdb_classifier with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Kicel/sparse_imdb_classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 1,614 Bytes
68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 68b0819 c11c2f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | ---
library_name: peft
license: mit
base_model: FacebookAI/xlm-roberta-base
tags:
- base_model:adapter:FacebookAI/xlm-roberta-base
- lora
- transformers
metrics:
- accuracy
model-index:
- name: sparse_imdb_classifier
results: []
---
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->
# sparse_imdb_classifier
This model is a fine-tuned version of [FacebookAI/xlm-roberta-base](https://huggingface.co/FacebookAI/xlm-roberta-base) on an unknown dataset.
It achieves the following results on the evaluation set:
- Loss: 0.3355
- Accuracy: 0.918
## 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: 5e-05
- train_batch_size: 16
- eval_batch_size: 16
- 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: 2
### Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
| 0.4666 | 1.0 | 1438 | 0.3793 | 0.913 |
| 0.3565 | 2.0 | 2876 | 0.3355 | 0.918 |
### Framework versions
- PEFT 0.17.1
- Transformers 4.57.1
- Pytorch 2.8.0+cu126
- Datasets 4.0.0
- Tokenizers 0.22.1 |