MODEL_CARD.md

# Model card - tox21_chemprop_classifier
### Model details
- Model name: Graph Isomorphism Network Tox21 Baseline
- Developer: MIT & Stanford (trained by JKU Linz)
- Paper URL: https://arxiv.org/abs/1810.00826
- Model type / architecture:
    - Graph Isomorphism Network implemented using PyTorch.
    - Hyperparameters: [link to config](https://huggingface.co/spaces/ml-jku/tox21_gin_classifier/blob/main/config/config.json)
    - A multitask network is trained for all Tox21 targets.
- Inference: Access via FastAPI endpoint. Upon a Tox21 prediction request, the model
generates and returns predictions for all Tox21 targets simultaneously.
- Model version: v0
- Model date: 14.10.2025
- Reproducibility: Code for full training is available and enables retraining of the model from
scratch.

### Intended use
This model serves as a baseline benchmark for evaluating and comparing toxicity prediction
methods across the 12 pathway assays of the Tox21 dataset. It is not intended for clinical
decision-making without experimental validation.

### Metric
Each Tox21 task is evaluated using the area under the receiver operating characteristic curve
(AUC). Overall performance is reported as the mean AUC across all individual tasks.

### Training data
Tox21 training and validation sets.

### Evaluation data
Tox21 test set.