antreaspiece's picture
Upload best classification model (flan-t5-small)
e0a38cb verified
---
license: apache-2.0
language: en
tags:
- temporal
- time-module
- pieces
- intent-classification
- temporal-intent
datasets:
- Pieces/temporal-intent-classification-dataset-split
base_model: google/flan-t5-small
metrics:
- accuracy
- f1
pipeline_tag: text-classification
---
# TIME-Module: Classification — flan-t5-small
## Model Description
Temporal intent classification model trained on the split dataset. Classifies user queries into 6 temporal intent categories: Action Scheduling, Content Retrieval, Current Status, Future Information/Planning, Non-Temporal, and Temporal - General.
## Training Details
- **Base Model:** [google/flan-t5-small](https://huggingface.co/google/flan-t5-small)
- **Architecture:** T5ForSequenceClassification
- **Dataset:** [Pieces/temporal-intent-classification-dataset-split](https://huggingface.co/datasets/Pieces/temporal-intent-classification-dataset-split) (15,488 train / 1,721 val / 4,303 test)
- **Training Steps:** 17,500
- **Learning Rate:** 3e-4
- **Batch Size:** 64 × 2
- **Mixed Precision:** bf16
- **Hardware:** NVIDIA RTX 4090 (24 GB)
## Results
| Metric | Value |
|--------|-------|
| accuracy | 95.91% |
| f1_weighted | ~95.9% |
## Usage
```python
from transformers import pipeline
pipe = pipeline('text-classification', model='Pieces/time-classification-flan-t5-small-split-best')
result = pipe('What meetings do I have tomorrow?')
```
## Part of the TIME-Module Project
This model is part of the TIME (Temporal Intent, Mapping, and Extraction) module, a suite of models for understanding and processing temporal information in natural language.
**Related models:**
- [Pieces/time-classification-flan-t5-small-split-best](https://huggingface.co/Pieces/time-classification-flan-t5-small-split-best) — Intent classification
- [Pieces/time-mapping-flan-t5-small-quality-best](https://huggingface.co/Pieces/time-mapping-flan-t5-small-quality-best) — Span prediction (best)
- [Pieces/time-mapping-t5gemma-270m-best](https://huggingface.co/Pieces/time-mapping-t5gemma-270m-best) — Span prediction (T5Gemma)
## Citation
```bibtex
@software{time_module,
title={TIME-Module: Temporal Intent, Mapping, and Extraction},
author={Pieces},
year={2026},
url={https://huggingface.co/Pieces}
}
```