| language: en | |
| tags: | |
| - transformers | |
| - text-classification | |
| - customer-feedback | |
| - aspect-based-sentiment-analysis | |
| pipeline_tag: text-classification | |
| # Aspect Detection Model | |
| This model is part of a customer feedback analysis project. | |
| ## Task | |
| Detect whether a customer review contains a target aspect or aspect category. | |
| ## Framework | |
| - Hugging Face Transformers | |
| - PyTorch | |
| ## Usage | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline( | |
| "text-classification", | |
| model="jiangzy1881/aspect-detection-model" | |
| ) | |
| classifier("The food was amazing but the service was slow.") | |
| ``` | |
| ## Project Structure | |
| This model was fine-tuned and saved from the assignment notebook workflow under: | |
| ```text | |
| DL_Project/models/ | |
| ``` | |