Finetuned DistilBERT for movie genre classification
This repository contains a finetuned DistilBERT utilizing MultiModalPredictor
on movie titles and genres.
Model
- Framework: AutoGluon MultiModal
- Image model library: TIMM
- Problem type: multiclass
- Target:
Genre - Selection metric: Balanced accuracy
- Validation balanced accuracy:
0.266667
Loading
from huggingface_hub import snapshot_download
from autogluon.multimodal import MultiModalPredictor
model_path = snapshot_download(
repo_id="eandujar/finetuned_model"
)
predictor = MultiModalPredictor.load(model_path)
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support