jiangzy1881's picture
Upload fine-tuned aspect detection model
634c13f verified
|
Raw
History Blame Contribute Delete
737 Bytes
metadata
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

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:

DL_Project/models/