--- language: en license: mit tags: - nutrition - behavior-change - tpb - ttm - distilbert - text-classification datasets: - custom metrics: - accuracy model-index: - name: franzzzzzzzzz/ttm-stage-nutrition results: - task: type: text-classification metrics: - type: accuracy value: 84.0 --- # 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](https://huggingface.co/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 ```python 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