franzzzzzzzzz/ttm-stage-nutrition
Model Description
Fine-tuned DistilBERT model for classifying stage of change (pre-contemplation, contemplation, preparation, action, maintenance) based on the Transtheoretical Model (TTM) for nutrition behavior change.
This model is a fine-tuned version of distilbert-base-uncased for behavior change inference in nutrition coaching contexts.
Training Data
- Training samples: 175
- Validation samples: 25
- Test samples: 50
- Total: 250 samples
Performance
- Test Accuracy: 84.0%
Intended Use
This model is designed for:
- Nutrition coaching chatbots
- Behavior change interventions
- Health psychology research
- Personalized dietary guidance
How to Use
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
import torch
# Load model
tokenizer = DistilBertTokenizer.from_pretrained("franzzzzzzzzz/ttm-stage-nutrition")
model = DistilBertForSequenceClassification.from_pretrained("franzzzzzzzzz/ttm-stage-nutrition")
# Predict
text = "I love healthy food, it's amazing!"
inputs = tokenizer(text, return_tensors='pt', padding=True, truncation=True)
outputs = model(**inputs)
prediction = torch.argmax(outputs.logits, dim=1).item()
# Convert 0-4 to 1-5 scale
score = prediction + 1
print(f"Score: {score}/5")
Limitations
- Trained on English text only
- Limited to nutrition/dietary contexts
- May not generalize to other health behaviors
- Requires context-appropriate input
Citation
If you use this model, please cite:
@misc{tpb-ttm-nutrition-models,
author = {Your Name},
title = {franzzzzzzzzz/ttm-stage-nutrition},
year = {2026},
publisher = {Hugging Face},
howpublished = {\url{https://huggingface.co/franzzzzzzzzz/ttm-stage-nutrition}}
}
License
MIT License
- Downloads last month
- 6
Evaluation results
- accuracyself-reported84.000