superhero-distilbert-predictor

superhero-distilbert-predictor

This model is a fine-tuned version of distilbert-base-uncased on the superhero-texts dataset. This model maps brief descriptions of popular superheroes to their respective comic book universes.

It achieves the following results on the evaluation set:

• Loss: 0.0161
• Accuracy: 1.0
• F1: 1.0
• Precision: 1.0
• Recall: 1.0

Intended uses & limitations

This model is strictly intended for educational use. Do not use this model to draw real world conclusions.

Training and evaluation data

This model was trained on an augmented set of 1100 synthetically generated superhero descriptions and their respective universe label. This model was validated against a set of 100 original, human curated descriptions.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 2e-05
• 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: 5

Training results

Framework versions

• Transformers 4.56.1
• Pytorch 2.8.0+cu126
• Datasets 4.0.0
• Tokenizers 0.22.0