Back to all models
fill-mask mask_token: [MASK]
Query this model
πŸ”₯ This model is currently loaded and running on the Inference API. ⚠️ This model could not be loaded by the inference API. ⚠️ This model can be loaded on the Inference API on-demand.
JSON Output
API endpoint  

⚑️ Upgrade your account to access the Inference API

Share Copied link to clipboard

Monthly model downloads

textattack/albert-base-v2-rotten_tomatoes textattack/albert-base-v2-rotten_tomatoes
last 30 days



Contributed by

3 team members Β· 84 models

How to use this model directly from the πŸ€—/transformers library:

Copy to clipboard
from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("textattack/albert-base-v2-rotten_tomatoes") model = AutoModelForMaskedLM.from_pretrained("textattack/albert-base-v2-rotten_tomatoes")
Uploaded in S3

albert-base-v2 fine-tuned with TextAttack on the rotten_tomatoes dataset

This `albert-base-v2` model was fine-tuned for sequence classificationusing TextAttack 
and the rotten_tomatoes dataset loaded using the `nlp` library. The model was fine-tuned 
for 10 epochs with a batch size of 128, a learning 
rate of 2e-05, and a maximum sequence length of 128. 
Since this was a classification task, the model was trained with a cross-entropy loss function. 
The best score the model achieved on this task was 0.8855534709193246, as measured by the 
eval set accuracy, found after 1 epoch.

For more information, check out [TextAttack on Github](