| ---
|
| 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
|
|
|
