metadata
tags:
- bertopic
library_name: bertopic
datasets:
- imdb
pipeline_tag: text-classification
language:
- en
imdb_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:
from bertopic import BERTopic
topic_model = BERTopic.load("davanstrien/imdb_bertopic")
topic_model.get_topic_info()
Topic overview
- Number of topics: 148
- Number of training documents: 103062
Click here for an overview of all topics.
| Topic ID | Topic Keywords | Topic Frequency | Label |
|---|---|---|---|
| -1 | film - movie - character - characters - very | 40 | -1_film_movie_character_characters |
| 0 | movie - film - movies - acting - actors | 25087 | 0_movie_film_movies_acting |
| 1 | episodes - shows - tv - television - watching | 2886 | 1_episodes_shows_tv_television |
| 2 | bollywood - kapoor - hindi - indian - rani | 1945 | 2_bollywood_kapoor_hindi_indian |
| 3 | horror - films - film - haunted - movies | 858 | 3_horror_films_film_haunted |
| 4 | takashi - samurai - manga - anime - films | 800 | 4_takashi_samurai_manga_anime |
| 5 | predator - alien - aliens - movie - film | 658 | 5_predator_alien_aliens_movie |
| 6 | amelie - cinema - films - film - movie | 652 | 6_amelie_cinema_films_film |
| 7 | nudity - porn - film - sexual - films | 647 | 7_nudity_porn_film_sexual |
| 8 | mexican - film - spanish - movie - mexico | 619 | 8_mexican_film_spanish_movie |
| 9 | comedies - comedy - humor - laughing - laughs | 589 | 9_comedies_comedy_humor_laughing |
| 10 | homosexuality - queer - homosexual - sexuality - gay | 562 | 10_homosexuality_queer_homosexual_sexuality |
| 11 | cary - cast - comedies - dunne - comedy | 459 | 11_cary_cast_comedies_dunne |
| 12 | film - wwii - movie - patton - war | 408 | 12_film_wwii_movie_patton |
| 13 | dracula - vampire - vampires - frankenstein - vlad | 405 | 13_dracula_vampire_vampires_frankenstein |
| 14 | zombie - zombies - romero - undead - horror | 402 | 14_zombie_zombies_romero_undead |
| 15 | kung - martial - chun - chan - fighter | 370 | 15_kung_martial_chun_chan |
| 16 | soviet - stalin - moscow - russian - russians | 355 | 16_soviet_stalin_moscow_russian |
| 17 | western - film - westerns - cast - wayne | 310 | 17_western_film_westerns_cast |
| 18 | blacks - slavery - racism - racial - slaves | 308 | 18_blacks_slavery_racism_racial |
| 19 | satanic - demon - satanist - devil - satan | 289 | 19_satanic_demon_satanist_devil |
| 20 | jurassic - prehistoric - lost - dinosaur - dinosaurs | 241 | 20_jurassic_prehistoric_lost_dinosaur |
| 21 | gameplay - rpg - games - adventure - game | 231 | 21_gameplay_rpg_games_adventure |
| 22 | daffy - looney - bunny - cartoon - cartoons | 228 | 22_daffy_looney_bunny_cartoon |
| 23 | bergman - films - swedish - henrik - film | 224 | 23_bergman_films_swedish_henrik |
| 24 | movie - dogs - animals - film - puppies | 216 | 24_movie_dogs_animals_film |
| 25 | hitchcock - thriller - suspense - film - films | 204 | 25_hitchcock_thriller_suspense_film |
| 26 | criminal - criminals - film - crime - cop | 199 | 26_criminal_criminals_film_crime |
| 27 | cinderella - disney - aladdin - fairy - anastasia | 191 | 27_cinderella_disney_aladdin_fairy |
| 28 | films - film - movie - racing - racer | 188 | 28_films_film_movie_racing |
| 29 | cinema - film - movie - actors - australian | 186 | 29_cinema_film_movie_actors |
| 30 | natives - tribe - tribes - film - navajo | 184 | 30_natives_tribe_tribes_film |
| 31 | jaws - shark - sharks - megalodon - movie | 174 | 31_jaws_shark_sharks_megalodon |
| 32 | holocaust - auschwitz - schindler - nazi - nazis | 173 | 32_holocaust_auschwitz_schindler_nazi |
| 33 | woodstock - concert - tour - documentary - stones | 170 | 33_woodstock_concert_tour_documentary |
| 34 | adoption - abortion - ending - movie - daughter | 166 | 34_adoption_abortion_ending_movie |
| 35 | bond - royale - connery - sutherland - craig | 161 | 35_bond_royale_connery_sutherland |
| 36 | zhang - chinese - beijing - wong - films | 154 | 36_zhang_chinese_beijing_wong |
| 37 | santa - scrooge - claus - christmas - carol | 152 | 37_santa_scrooge_claus_christmas |
| 38 | messiah - biblical - gospel - jesus - film | 149 | 38_messiah_biblical_gospel_jesus |
| 39 | sandler - gilmore - adam - comedies - comedy | 145 | 39_sandler_gilmore_adam_comedies |
| 40 | woody - allen - wilder - bogart - movies | 142 | 40_woody_allen_wilder_bogart |
| 41 | pacino - film - movie - actor - al | 142 | 41_pacino_film_movie_actor |
| 42 | novel - book - books - movie - reading | 140 | 42_novel_book_books_movie |
| 43 | yankee - yankees - baseball - ruth - 1961 | 138 | 43_yankee_yankees_baseball_ruth |
| 44 | musicals - astaire - roberta - musical - irene | 136 | 44_musicals_astaire_roberta_musical |
| 45 | gangster - gangsters - cagney - mafia - gangs | 135 | 45_gangster_gangsters_cagney_mafia |
| 46 | actress - joan - crawford - film - movie | 134 | 46_actress_joan_crawford_film |
| 47 | italian - italians - sicily - cinema - films | 127 | 47_italian_italians_sicily_cinema |
| 48 | holmes - sherlock - doyle - watson - hound | 124 | 48_holmes_sherlock_doyle_watson |
| 49 | slashers - movie - film - slasher - movies | 117 | 49_slashers_movie_film_slasher |
| 50 | romance - movie - comedy - movies - romantic | 116 | 50_romance_movie_comedy_movies |
| 51 | hogan - daredevil - hulk - daredevils - movies | 114 | 51_hogan_daredevil_hulk_daredevils |
| 52 | werewolf - werewolves - wolf - wolfen - movie | 112 | 52_werewolf_werewolves_wolf_wolfen |
| 53 | robin - flynn - films - hood - film | 110 | 53_robin_flynn_films_hood |
| 54 | agatha - novels - novel - cast - marple | 108 | 54_agatha_novels_novel_cast |
| 55 | netforce - supercomputer - security - hacking - computers | 104 | 55_netforce_supercomputer_security_hacking |
| 56 | dragonball - goku - dragon - dragonheart - dragons | 104 | 56_dragonball_goku_dragon_dragonheart |
| 57 | pirates - pirate - blackbeard - captain - film | 102 | 57_pirates_pirate_blackbeard_captain |
| 58 | dancing - dance - dancer - choreography - dancers | 97 | 58_dancing_dance_dancer_choreography |
| 59 | conspiracy - 911 - documentary - documentaries - pentagon | 94 | 59_conspiracy_911_documentary_documentaries |
| 60 | chainsaw - massacre - slaughterhouse - movie - movies | 94 | 60_chainsaw_massacre_slaughterhouse_movie |
| 61 | musical - soundtrack - film - movie - music | 93 | 61_musical_soundtrack_film_movie |
| 62 | manhattan - burns - woody - films - sidewalks | 90 | 62_manhattan_burns_woody_films |
| 63 | afghan - taliban - osama - film - movie | 89 | 63_afghan_taliban_osama_film |
| 64 | turkish - turk - turks - turkey - kurdish | 89 | 64_turkish_turk_turks_turkey |
| 65 | cannibalism - cannibal - cannibals - slaughterhouse - holocaust | 84 | 65_cannibalism_cannibal_cannibals_slaughterhouse |
| 66 | movie - movies - kids - film - children | 82 | 66_movie_movies_kids_film |
| 67 | hellraiser - hellraisers - hellworld - hellbound - hellboy | 82 | 67_hellraiser_hellraisers_hellworld_hellbound |
| 68 | korean - korea - seoul - cinema - film | 81 | 68_korean_korea_seoul_cinema |
| 69 | jedi - darth - prequels - skywalker - lightsaber | 81 | 69_jedi_darth_prequels_skywalker |
| 70 | eddie - sequel - beverly - murphy - cop | 80 | 70_eddie_sequel_beverly_murphy |
| 71 | fiona - irish - donegal - ireland - dublin | 79 | 71_fiona_irish_donegal_ireland |
| 72 | beatles - ringo - mccartney - lennon - concert | 79 | 72_beatles_ringo_mccartney_lennon |
| 73 | ninja - ninjas - kung - martial - cobra | 79 | 73_ninja_ninjas_kung_martial |
| 74 | murder - prosecutor - innocent - justice - courtroom | 77 | 74_murder_prosecutor_innocent_justice |
| 75 | lynch - mulholland - films - film - eraserhead | 76 | 75_lynch_mulholland_films_film |
| 76 | godzilla - monster - monsters - shinoda - tokyo | 76 | 76_godzilla_monster_monsters_shinoda |
| 77 | scorsese - film - films - comedy - goodfellas | 74 | 77_scorsese_film_films_comedy |
| 78 | clancy - ryan - cia - director - thriller | 74 | 78_clancy_ryan_cia_director |
| 79 | ringu - ring - prequel - sequel - film | 72 | 79_ringu_ring_prequel_sequel |
| 80 | candyman - candy - tale - horror - movie | 70 | 80_candyman_candy_tale_horror |
| 81 | guns - gun - shooting - bullets - lethal | 70 | 81_guns_gun_shooting_bullets |
| 82 | claudius - roman - romans - emperors - emperor | 69 | 82_claudius_roman_romans_emperors |
| 83 | jason - friday - 13th - movie - sequel | 69 | 83_jason_friday_13th_movie |
| 84 | yugoslav - yugoslavian - bosnians - bosnian - croatians | 69 | 84_yugoslav_yugoslavian_bosnians_bosnian |
| 85 | giallo - gialli - giallos - thriller - film | 68 | 85_giallo_gialli_giallos_thriller |
| 86 | arthurian - arthur - merlin - medieval - excalibur | 68 | 86_arthurian_arthur_merlin_medieval |
| 87 | robot - robots - robocop - asimov - film | 67 | 87_robot_robots_robocop_asimov |
| 88 | thai - thailand - movie - bangkok - jungle | 67 | 88_thai_thailand_movie_bangkok |
| 89 | batman - gotham - joker - superhero - comic | 67 | 89_batman_gotham_joker_superhero |
| 90 | chaplin - comedies - films - film - circus | 67 | 90_chaplin_comedies_films_film |
| 91 | superman - lois - reeve - clark - krypton | 67 | 91_superman_lois_reeve_clark |
| 92 | marvel - wolverine - superhero - comics - cyclops | 66 | 92_marvel_wolverine_superhero_comics |
| 93 | austen - novel - prejudice - emma - firth | 65 | 93_austen_novel_prejudice_emma |
| 94 | movie - film - movies - future - time | 64 | 94_movie_film_movies_future |
| 95 | doris - actress - ruth - judy - hollywood | 64 | 95_doris_actress_ruth_judy |
| 96 | vhs - dvd - movie - film - movies | 63 | 96_vhs_dvd_movie_film |
| 97 | corpses - flesh - cadavers - bodies - snatcher | 63 | 97_corpses_flesh_cadavers_bodies |
| 98 | ants - ant - movie - wasp - wasps | 62 | 98_ants_ant_movie_wasp |
| 99 | apostle - sermon - actor - duvall - preacher | 60 | 99_apostle_sermon_actor_duvall |
| 100 | twins - twin - olsen - actresses - movie | 60 | 100_twins_twin_olsen_actresses |
| 101 | tudors - tudor - anne - elizabeth - boleyn | 59 | 101_tudors_tudor_anne_elizabeth |
| 102 | rocky - movie - film - tommy - rusty | 59 | 102_rocky_movie_film_tommy |
| 103 | seagal - movie - movies - scenes - mercenary | 56 | 103_seagal_movie_movies_scenes |
| 104 | film - documentary - witch - movies - movie | 56 | 104_film_documentary_witch_movies |
| 105 | panther - clouseau - films - film - movies | 55 | 105_panther_clouseau_films_film |
| 106 | van - cyborg - movie - film - arquette | 55 | 106_van_cyborg_movie_film |
| 107 | freddy - nightmare - elm - horror - sequel | 54 | 107_freddy_nightmare_elm_horror |
| 108 | crow - movie - film - films - angels | 53 | 108_crow_movie_film_films |
| 109 | bambi - sequels - sequel - disney - movie | 53 | 109_bambi_sequels_sequel_disney |
| 110 | potter - narnia - hogwarts - harry - dumbledore | 53 | 110_potter_narnia_hogwarts_harry |
| 111 | puppetmaster - puppet - puppets - curse - magrew | 51 | 111_puppetmaster_puppet_puppets_curse |
| 112 | germans - german - cinema - film - hans | 51 | 112_germans_german_cinema_film |
| 113 | eastwood - harry - callahan - cop - dirty | 51 | 113_eastwood_harry_callahan_cop |
| 114 | bruce - willis - movies - film - mcclane | 51 | 114_bruce_willis_movies_film |
| 115 | dutch - dutchman - amsterdam - netherlands - holland | 50 | 115_dutch_dutchman_amsterdam_netherlands |
| 116 | highlander - macleod - sequel - immortals - movie | 50 | 116_highlander_macleod_sequel_immortals |
| 117 | clerks - comedy - film - movies - movie | 50 | 117_clerks_comedy_film_movies |
| 118 | metalhead - metallica - metal - genres - genre | 50 | 118_metalhead_metallica_metal_genres |
| 119 | angelopoulos - greek - trilogy - greece - tragedy | 50 | 119_angelopoulos_greek_trilogy_greece |
| 120 | wilder - pryor - comedy - comedian - comedians | 49 | 120_wilder_pryor_comedy_comedian |
| 121 | sopranos - soprano - tony - godfather - mafia | 48 | 121_sopranos_soprano_tony_godfather |
| 122 | dracula - frankenstein - cat - poelzig - horror | 48 | 122_dracula_frankenstein_cat_poelzig |
| 123 | myers - comedy - mike - guru - wayne | 48 | 123_myers_comedy_mike_guru |
| 124 | comedy - lemmon - movie - cast - neil | 47 | 124_comedy_lemmon_movie_cast |
| 125 | jezebel - scarlett - actress - warner - film | 47 | 125_jezebel_scarlett_actress_warner |
| 126 | cast - crystal - mafia - comedies - movie | 47 | 126_cast_crystal_mafia_comedies |
| 127 | marijuana - drug - pot - weed - marihuana | 47 | 127_marijuana_drug_pot_weed |
| 128 | autism - autistic - charlie - movie - sam | 46 | 128_autism_autistic_charlie_movie |
| 129 | sergio - corbucci - franco - mercenary - italian | 45 | 129_sergio_corbucci_franco_mercenary |
| 130 | actress - acting - audrey - hollywood - griffith | 45 | 130_actress_acting_audrey_hollywood |
| 131 | columbo - murderer - murder - detective - movie | 44 | 131_columbo_murderer_murder_detective |
| 132 | egyptian - mohamed - egyptians - egypt - ahmed | 44 | 132_egyptian_mohamed_egyptians_egypt |
| 133 | che - castro - ernesto - cuban - havana | 44 | 133_che_castro_ernesto_cuban |
| 134 | horse - horses - thoroughbred - stallion - dakota | 44 | 134_horse_horses_thoroughbred_stallion |
| 135 | costello - abbott - comedy - comedians - films | 43 | 135_costello_abbott_comedy_comedians |
| 136 | conchita - surrealistic - surrealism - actresses - surrealist | 43 | 136_conchita_surrealistic_surrealism_actresses |
| 137 | lindsay - movie - actress - film - teenage | 43 | 137_lindsay_movie_actress_film |
| 138 | mummy - mummies - tomb - archaeologist - film | 42 | 138_mummy_mummies_tomb_archaeologist |
| 139 | ebola - virus - movie - epidemic - film | 42 | 139_ebola_virus_movie_epidemic |
| 140 | anaïs - anais - erotica - film - fiction | 42 | 140_anaïs_anais_erotica_film |
| 141 | mormon - lds - mormons - brigham - missionaries | 42 | 141_mormon_lds_mormons_brigham |
| 142 | chuck - chucks - norris - karate - fights | 42 | 142_chuck_chucks_norris_karate |
| 143 | beowulf - grendel - hrothgar - siegfried - gaiman | 41 | 143_beowulf_grendel_hrothgar_siegfried |
| 144 | martin - rigby - comedy - film - noir | 41 | 144_martin_rigby_comedy_film |
| 145 | troma - tromeo - keoma - film - movie | 41 | 145_troma_tromeo_keoma_film |
| 146 | airplane - passengers - airlines - plane - pilot | 41 | 146_airplane_passengers_airlines_plane |
Training hyperparameters
- calculate_probabilities: False
- language: None
- low_memory: False
- min_topic_size: 40
- n_gram_range: (1, 1)
- nr_topics: None
- seed_topic_list: None
- top_n_words: 10
- verbose: False
Framework versions
- Numpy: 1.22.4
- HDBSCAN: 0.8.29
- UMAP: 0.5.3
- Pandas: 1.5.3
- Scikit-Learn: 1.2.2
- Sentence-transformers: 2.2.2
- Transformers: 4.29.2
- Numba: 0.56.4
- Plotly: 5.13.1
- Python: 3.10.11