CoolFace
Modelpublic

enesgerem/airline-sentiment-models

sourceHugging Facecc-by-nc-sa-4.0updated 4mo agoView on Hugging Face
0likes
Model Card

Airline Passenger Sentiment Models

Fine-tuned transformer checkpoints from the M.Sc. thesis "Multi-Source Natural Language Processing for Airline Passenger Sentiment and Satisfaction Analysis" (Enes Gerem, Hacettepe University).

Each checkpoint is a 3-class sentiment classifier (negative / neutral / positive) fine-tuned on one of seven source configurations drawn from Twitter, Skytrax, and TripAdvisor airline feedback.

Label mapping

idlabel
0negative
1neutral
2positive

Available checkpoints (subfolders)

FamilySource configSubfolder
BERT-basetwitterbert/twitter
BERT-baseskytraxbert/skytrax
BERT-basetripadvisorbert/tripadvisor
BERT-basetwitter+skytraxbert/twitter+skytrax
BERT-basetwitter+tripadvisorbert/twitter+tripadvisor
BERT-baseskytrax+tripadvisorbert/skytrax+tripadvisor
BERT-basetwitter+skytrax+tripadvisorbert/twitter+skytrax+tripadvisor
RoBERTa-basetwitterroberta/twitter
RoBERTa-baseskytraxroberta/skytrax
RoBERTa-basetripadvisorroberta/tripadvisor
RoBERTa-basetwitter+skytraxroberta/twitter+skytrax
RoBERTa-basetwitter+tripadvisorroberta/twitter+tripadvisor
RoBERTa-baseskytrax+tripadvisorroberta/skytrax+tripadvisor
RoBERTa-basetwitter+skytrax+tripadvisorroberta/twitter+skytrax+tripadvisor
DistilBERT-basetwitterdistilbert/twitter
DistilBERT-baseskytraxdistilbert/skytrax
DistilBERT-basetripadvisordistilbert/tripadvisor
DistilBERT-basetwitter+skytraxdistilbert/twitter+skytrax
DistilBERT-basetwitter+tripadvisordistilbert/twitter+tripadvisor
DistilBERT-baseskytrax+tripadvisordistilbert/skytrax+tripadvisor
DistilBERT-basetwitter+skytrax+tripadvisordistilbert/twitter+skytrax+tripadvisor

Usage

python
from transformers import AutoTokenizer, AutoModelForSequenceClassification

repo = "enesgerem/airline-sentiment-models"
subfolder = "roberta/twitter"   # pick any subfolder from the table above

tokenizer = AutoTokenizer.from_pretrained(repo, subfolder=subfolder)
model = AutoModelForSequenceClassification.from_pretrained(repo, subfolder=subfolder)

Citation / data

Datasets are used for non-commercial academic research. The Twitter component is the Crowdflower US Airline Sentiment benchmark (CC BY-NC-SA); the Skytrax and TripAdvisor components are point-in-time scrapes of publicly available review pages, used under each platform's terms of service.