Team Ai
Modelpublic

sneakykilli/Topic_Modelling_Airlines_BERTopic

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes6downloads
Model Card

tags:

  • —bertopic libraryname: bertopic pipelinetag: text-classification ---

TopicModellingAirlines_BERTopic

This is a BERTopic model. BERTopic is a flexible and modular topic modeling framework that allows for the generation of easily interpretable topics from large datasets.

Usage

To use this model, please install BERTopic:

pip install -U bertopic

You can use the model as follows:

python
from bertopic import BERTopic
topic_model = BERTopic.load("sneakykilli/Topic_Modelling_Airlines_BERTopic")

topic_model.get_topic_info()

Topic overview

  • —Number of topics: 17
  • —Number of training documents: 5134

<details> <summary>Click here for an overview of all topics.</summary>

Topic IDTopic KeywordsTopic FrequencyLabel
-1killiair - flight - service - customer - airport23-1killiairflightservicecustomer
0killiair - doha - flight - service - worst2399poorcustomerexperience
1bag - luggage - cabin - bags - pay639luggage_fee
2flight - delayed - hours - delay - killiair386delays
3check - ryan - online - air - killiair334checkinprocess
4refund - killiair - flight - cancelled - booking293refund
5jet - easy - flight - cancelled - refund237refundcancelledflights
6seats - seat - plane - flight - killiair227inflight_facilities
7luggage - lost - bag - killiair - baggage154luggage_lost
8holiday - holidays - hotel - killiair - booked102hotel
9thank - amazing - crew - flight - thanks81goodcustomerexperience
10change - price - 115 - fare - booking59changeticketfee
11food - meal - dubai - flight - killiair48inflight_service
12car - hire - rental - insurance - card47car
13seats - seat - paid - extra - window41seating_fees
14service - killiair - customer - zero - customers37poorcustomerexperience
15stansted - flight - airport - parking - killiair27airport_facilities

</details>

Training hyperparameters

  • —calculate_probabilities: False
  • —language: None
  • —low_memory: False
  • —mintopicsize: 10
  • —ngramrange: (1, 1)
  • —nr_topics: None
  • —seedtopiclist: None
  • —topnwords: 10
  • —verbose: False
  • —zeroshotminsimilarity: 0.7
  • —zeroshottopiclist: None

Framework versions

  • —Numpy: 1.24.3
  • —HDBSCAN: 0.8.33
  • —UMAP: 0.5.5
  • —Pandas: 2.0.3
  • —Scikit-Learn: 1.2.2
  • —Sentence-transformers: 2.3.1
  • —Transformers: 4.36.2
  • —Numba: 0.57.1
  • —Plotly: 5.16.1
  • —Python: 3.10.12