Model: parlange/mlp-mixer-gravit-s2
Author: parlange
Pipeline: image-classification
Library: timm
Tags: timm, pytorch, safetensors, vision-transformer, image-classification, mlp-mixer, gravitational-lensing, strong-lensing, astronomy, astrophysics, dataset:parlange/gravit-c21, arxiv:2509.00226, license:apache-2.0, model-index, region:us
Card metadata: license: apache-2.0
tags: vision-transformer, image-classification, pytorch, timm, mlp-mixer, gravitational-lensing, strong-lensing, astronomy, astrophysics
datasets: parlange/gravit-c21
metrics: accuracy, auc, f1
paper: title: GraViT: A Gravitational Lens Discovery Toolkit with Vision Transformers
url: https://arxiv.org/abs/2509.00226
authors: Parlange et al.
model-index: name: MLP-Mixer-s2
results: task: type: image-classification
name: Strong Gravitational Lens Discovery
dataset: type: common-test-sample
name: Common Test Sample (More et al. 2024)
metrics: type: accuracy
value: 0.8183
name: Average Accuracy
verified: no, type: auc
value: 0.8622
name: Average AUC-ROC
verified: no, type: f1
value: 0.5446
name: Average F1-Score
verified: no
Likes: 0 | All-time downloads: 39
Created: 2025-09-06 22:08:57