Downloads · 30 days
0
toxicthesis-gemini-3.5-flash-length-classification-5
simocorbo/toxicthesis-gemini-3.5-flash-length-classification-5
toxicthesis-gemini-3.5-flash-length-classification-5 is a text classification model from simocorbo. Use it when you need a label for a piece of text. It is set up for pytorch. The card lists the license as mit.
Toxicity prediction model trained on the GEMINI-3.5-FLASH dataset.
Access
Public
Updated Jun 12, 2026
Repo size
4.2 KB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.csv651 KB · 99%
At a glance
- Task
- Text Classification
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Jun 12, 2026
- Updated
- Jun 12, 2026
- SHA
- ca2d68e2
Try a prompt
Model card
From the Hugging Face model README
Length Baseline - GEMINI-3.5-FLASH - Classification (5 classes)
Toxicity prediction model trained on the GEMINI-3.5-FLASH dataset.
| Property | Value |
|---|---|
| Model | Length Baseline |
| Task | Classification (5 classes) |
| Dataset | gemini-3.5-flash |
| Framework | PyTorch / PyTorch Lightning |
Model Information
See the ToxicThesis repository for model class documentation.
Usage
from huggingface_hub import hf_hub_download
import torch
checkpoint_path = hf_hub_download(
repo_id="simocorbo/toxicthesis-gemini-3.5-flash-length-classification-5",
filename="checkpoints/best.pt"
)
checkpoint = torch.load(checkpoint_path, map_location='cpu', weights_only=False)
print("Checkpoint keys:", checkpoint.keys())
# See ToxicThesis repository for model implementation
# git clone https://github.com/simo-corbo/ToxicThesis
Score Interpretation
| Output | Range | Meaning |
|---|---|---|
probabilities | List[float] | Probability distribution over 5 classes. |
class | 0 to 4 | Predicted class (argmax of probabilities). |
Classes: 5 toxicity levels, where higher class index = more toxic.
Files
| File | Description |
|---|---|
checkpoints/best.pt | Model checkpoint (best validation loss) |
hparams.yaml | Hyperparameters used for training |
train.csv | Training metrics per epoch |
val.csv | Validation metrics per epoch |
vocab_stanza_hybrid.pkl | Vocabulary (for tree-based models) |
Installation
# Clone ToxicThesis for full model implementations
git clone https://github.com/simo-corbo/ToxicThesis
cd ToxicThesis
pip install -r requirements.txt
# Or install dependencies directly
pip install torch transformers huggingface_hub fasttext-wheel stanza
Citation
@software{toxicthesis2025,
title={ToxicThesis},
author={Corbo, Simone},
year={2025},
url={https://github.com/simo-corbo/ToxicThesis}
}
- Task
- Text Classification
- Library
- pytorch
- License
- mit
- Languages
- en
- Created
- Jun 12, 2026
- Updated
- Jun 12, 2026