xsum_6789_5000000_2500000_v1_150topics_train

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("KingKazma/xsum_6789_5000000_2500000_v1_150topics_train")

topic_model.get_topic_info()

Topic overview

  • Number of topics: 150
  • Number of training documents: 204045
Click here for an overview of all topics.
Topic ID Topic Keywords Topic Frequency Label
-1 said - year - one - mr - also 5 -1_said_year_one_mr
0 half - win - kick - foul - game 120934 0_half_win_kick_foul
1 labour - eu - party - uk - vote 12329 1_labour_eu_party_uk
2 race - champion - world - title - fight 7075 2_race_champion_world_title
3 school - education - health - patient - nhs 5962 3_school_education_health_patient
4 police - murder - court - man - mr 4410 4_police_murder_court_man
5 rail - train - transport - driver - road 4066 5_rail_train_transport_driver
6 al - syria - syrian - attack - us 3623 6_al_syria_syrian_attack
7 prison - police - inquiry - abuse - court 3574 7_prison_police_inquiry_abuse
8 africa - african - nigeria - president - boko 2827 8_africa_african_nigeria_president
9 fire - blaze - building - firefighter - smoke 1675 9_fire_blaze_building_firefighter
10 russia - russian - ukraine - putin - president 1460 10_russia_russian_ukraine_putin
11 coastguard - fish - water - rescue - sea 1341 11_coastguard_fish_water_rescue
12 dog - animal - zoo - cat - rspca 1312 12_dog_animal_zoo_cat
13 sale - company - share - firm - apple 1302 13_sale_company_share_firm
14 data - security - facebook - user - google 1224 14_data_security_facebook_user
15 energy - oil - gas - climate - price 1084 15_energy_oil_gas_climate
16 space - earth - planet - mission - satellite 1072 16_space_earth_planet_mission
17 cuba - venezuela - president - farc - maduro 1057 17_cuba_venezuela_president_farc
18 art - council - library - building - museum 1040 18_art_council_library_building
19 ireland - dup - northern - sinn - féin 1020 19_ireland_dup_northern_sinn
20 bank - tax - fraud - money - bitcoin 1008 20_bank_tax_fraud_money
21 flood - flooding - water - rain - storm 959 21_flood_flooding_water_rain
22 trump - clinton - mr - republican - president 955 22_trump_clinton_mr_republican
23 film - actor - star - show - best 920 23_film_actor_star_show
24 plane - aircraft - flight - drone - pilot 918 24_plane_aircraft_flight_drone
25 korea - north - china - korean - kim 867 25_korea_north_china_korean
26 music - album - song - band - singer 839 26_music_album_song_band
27 virus - ebola - vaccine - infection - disease 827 27_virus_ebola_vaccine_infection
28 abortion - hospital - care - death - baby 765 28_abortion_hospital_care_death
29 rate - growth - economy - inflation - bank 759 29_rate_growth_economy_inflation
30 airport - airline - heathrow - passenger - runway 727 30_airport_airline_heathrow_passenger
31 yn - mae - ar - wedi - ei 653 31_yn_mae_ar_wedi
32 thailand - myanmar - thai - kyi - suu 560 32_thailand_myanmar_thai_kyi
33 attack - paris - french - france - police 542 33_attack_paris_french_france
34 migrant - refugee - asylum - eu - europe 535 34_migrant_refugee_asylum_eu
35 ship - ferry - vessel - hms - navy 522 35_ship_ferry_vessel_hms
36 drug - cannabis - heroin - drugs - cocaine 466 36_drug_cannabis_heroin_drugs
37 doping - athlete - iaaf - olympic - wada 449 37_doping_athlete_iaaf_olympic
38 bbc - 00 - radio - tv - news 442 38_bbc_00_radio_tv
39 pte - soldier - cpl - war - inquest 422 39_pte_soldier_cpl_war
40 india - modi - delhi - bjp - indian 415 40_india_modi_delhi_bjp
41 bird - bee - rspb - wildlife - birds 398 41_bird_bee_rspb_wildlife
42 smoking - pollution - air - tobacco - cigarettes 394 42_smoking_pollution_air_tobacco
43 church - bishop - pope - vatican - cardinal 393 43_church_bishop_pope_vatican
44 greece - eurozone - greek - bank - ecb 381 44_greece_eurozone_greek_bank
45 museum - roman - archaeologist - site - found 335 45_museum_roman_archaeologist_site
46 alcohol - drink - whisky - drinking - wine 320 46_alcohol_drink_whisky_drinking
47 australia - labor - nauru - australian - turnbull 307 47_australia_labor_nauru_australian
48 car - vw - emission - volkswagen - diesel 299 48_car_vw_emission_volkswagen
49 unsupported - updated - playback - device - media 294 49_unsupported_updated_playback_device
50 robot - car - vehicle - technology - computer 293 50_robot_car_vehicle_technology
51 broadband - bt - ofcom - mobile - openreach 257 51_broadband_bt_ofcom_mobile
52 prince - queen - duchess - royal - duke 253 52_prince_queen_duchess_royal
53 school - pupil - bullying - teacher - student 250 53_school_pupil_bullying_teacher
54 festival - event - yorkshire - edinburgh - tour 241 54_festival_event_yorkshire_edinburgh
55 updated - gmt - bst - 2017 - last 240 55_updated_gmt_bst_2017
56 fifa - blatter - platini - football - swiss 225 56_fifa_blatter_platini_football
57 food - product - meat - horsemeat - nestle 224 57_food_product_meat_horsemeat
58 game - console - nintendo - vr - oculus 209 58_game_console_nintendo_vr
59 book - novel - author - prize - writer 208 59_book_novel_author_prize
60 welsh - language - ashley - sports - sport 201 60_welsh_language_ashley_sports
61 manchester - concert - abedi - grande - ariana 199 61_manchester_concert_abedi_grande
62 rousseff - lula - brazil - petrobras - temer 196 62_rousseff_lula_brazil_petrobras
63 charity - kids - batmanghelidjh - fundraising - service 185 63_charity_kids_batmanghelidjh_fundraising
64 mountain - avalanche - everest - climber - snow 185 64_mountain_avalanche_everest_climber
65 fgm - trafficking - slavery - sex - victim 180 65_fgm_trafficking_slavery_sex
66 trade - tpp - obamacare - ceta - us 159 66_trade_tpp_obamacare_ceta
67 assange - wikileaks - snowden - manning - ecuador 152 67_assange_wikileaks_snowden_manning
68 trident - defence - nuclear - submarine - army 145 68_trident_defence_nuclear_submarine
69 berlusconi - italy - italian - regeni - renzi 135 69_berlusconi_italy_italian_regeni
70 choi - park - ms - samsung - lee 128 70_choi_park_ms_samsung
71 mayor - ford - council - rahman - devolution 126 71_mayor_ford_council_rahman
72 mine - miner - coal - mining - pit 126 72_mine_miner_coal_mining
73 ice - antarctic - sea - arctic - ocean 126 73_ice_antarctic_sea_arctic
74 transgender - abortion - law - gay - marriage 121 74_transgender_abortion_law_gay
75 gambling - lottery - jackpot - ticket - betting 120 75_gambling_lottery_jackpot_ticket
76 gay - marriage - same - sex - ashers 119 76_gay_marriage_same_sex
77 flag - confederate - white - university - statue 115 77_flag_confederate_white_university
78 fossil - dinosaur - human - modern - homo 109 78_fossil_dinosaur_human_modern
79 science - research - graphene - prof - prize 109 79_science_research_graphene_prof
80 nuclear - japan - reactor - radiation - plant 106 80_nuclear_japan_reactor_radiation
81 tree - ash - trees - woodland - dieback 105 81_tree_ash_trees_woodland
82 trump - golf - crofting - turnberry - hie 104 82_trump_golf_crofting_turnberry
83 doping - armstrong - cycling - sharapova - meldonium 101 83_doping_armstrong_cycling_sharapova
84 picture - scotlandpictures - bbcscotlandpics - photo - selection 100 84_picture_scotlandpictures_bbcscotlandpics_photo
85 water - utilities - customer - cryptosporidium - company 89 85_water_utilities_customer_cryptosporidium
86 christmas - cake - santa - going - toy 89 86_christmas_cake_santa_going
87 fashion - dress - vogue - wear - designer 82 87_fashion_dress_vogue_wear
88 nba - lakers - cavaliers - warriors - bryant 82 88_nba_lakers_cavaliers_warriors
89 woman - women - female - gender - 100 74 89_woman_women_female_gender
90 robot - ant - prof - insect - robots 74 90_robot_ant_prof_insect
91 cosby - constand - mueller - swift - lawyer 68 91_cosby_constand_mueller_swift
92 note - polymer - bank - banknote - coin 66 92_note_polymer_bank_banknote
93 argentina - argentine - falklands - falkland - islands 63 93_argentina_argentine_falklands_falkland
94 suicide - happiness - expectancy - life - age 62 94_suicide_happiness_expectancy_life
95 volcano - ash - eruption - earthquake - bgs 61 95_volcano_ash_eruption_earthquake
96 nfl - patriots - brady - quarterback - bowl 60 96_nfl_patriots_brady_quarterback
97 light - lighting - sky - dark - clock 60 97_light_lighting_sky_dark
98 ring - diamond - jewellery - carat - jewel 52 98_ring_diamond_jewellery_carat
99 pipeline - dakota - oil - sioux - keystone 49 99_pipeline_dakota_oil_sioux
100 bell - minster - ringers - ringing - imber 49 100_bell_minster_ringers_ringing
101 mail - royal - post - parcel - delivery 47 101_mail_royal_post_parcel
102 breastfeeding - advert - body - asa - image 47 102_breastfeeding_advert_body_asa
103 sleep - clock - screen - hour - internet 46 103_sleep_clock_screen_hour
104 island - islands - prime - becomes - minister 44 104_island_islands_prime_becomes
105 qatar - worker - garment - factory - bangladesh 43 105_qatar_worker_garment_factory
106 laser - weapon - firearm - pilot - airgun 41 106_laser_weapon_firearm_pilot
107 palmyra - ancient - syrian - site - cultural 39 107_palmyra_ancient_syrian_site
108 pokemon - game - ai - go - alphago 39 108_pokemon_game_ai_go
109 flag - emojis - emoji - unicode - design 37 109_flag_emojis_emoji_unicode
110 takata - airbags - recall - airbag - honda 37 110_takata_airbags_recall_airbag
111 clown - clowns - craze - creepy - dressed 35 111_clown_clowns_craze_creepy
112 india - film - indian - bollywood - food 34 112_india_film_indian_bollywood
113 airbnb - bite - homeless - social - littlejohn 33 113_airbnb_bite_homeless_social
114 follow - - - - 33 114_follow___
115 quantum - computing - photon - qubits - computer 29 115_quantum_computing_photon_qubits
116 bundy - federal - refuge - rancher - sweat 26 116_bundy_federal_refuge_rancher
117 cubs - baseball - series - curse - mets 25 117_cubs_baseball_series_curse
118 india - indian - china - modi - chinese 24 118_india_indian_china_modi
119 sinkhole - hole - florida - fontmell - collapse 24 119_sinkhole_hole_florida_fontmell
120 pitch - fixture - referee - rain - postponed 24 120_pitch_fixture_referee_rain
121 bloodhound - speed - car - record - rocket 24 121_bloodhound_speed_car_record
122 malala - pakistan - yousafzai - taliban - education 24 122_malala_pakistan_yousafzai_taliban
123 depp - joyce - dog - australia - boo 22 123_depp_joyce_dog_australia
124 list - managerial - below - league - by 18 124_list_managerial_below_league
125 warrior - joint - unmanned - exercise - navy 17 125_warrior_joint_unmanned_exercise
126 polanski - extradition - polish - geimer - poland 17 126_polanski_extradition_polish_geimer
127 deaf - language - bsl - sign - hearing 16 127_deaf_language_bsl_sign
128 name - boys - girls - boy - top 15 128_name_boys_girls_boy
129 poppy - war - weeping - cummins - 888 15 129_poppy_war_weeping_cummins
130 tattoo - tattooists - tattooing - piercing - tattooist 15 130_tattoo_tattooists_tattooing_piercing
131 balakrishnan - commune - cult - aravindan - woman 13 131_balakrishnan_commune_cult_aravindan
132 visa - brain - cothill - mrs - family 13 132_visa_brain_cothill_mrs
133 battery - lithium - ion - batteries - cathode 13 133_battery_lithium_ion_batteries
134 peachey - tip - calendar - finance - updated 12 134_peachey_tip_calendar_finance
135 emperor - akihito - prince - throne - imperial 12 135_emperor_akihito_prince_throne
136 dog - personality - animal - wolf - human 11 136_dog_personality_animal_wolf
137 decay - dental - tooth - teeth - oral 11 137_decay_dental_tooth_teeth
138 puerto - princess - urdangarin - rico - cristina 10 138_puerto_princess_urdangarin_rico
139 pitt - jolie - paltrow - married - divorce 10 139_pitt_jolie_paltrow_married
140 minnock - ethan - wildblood - father - court 8 140_minnock_ethan_wildblood_father
141 breaking - bbcbreaking - alert - fullest - via 8 141_breaking_bbcbreaking_alert_fullest
142 balcony - berkeley - irish - student - donohoe 7 142_balcony_berkeley_irish_student
143 flower - garden - flowered - botanic - arum 6 143_flower_garden_flowered_botanic
144 potter - beasts - grindelwald - dumbledore - film 6 144_potter_beasts_grindelwald_dumbledore
145 canal - waterway - oakham - towpath - lock 5 145_canal_waterway_oakham_towpath
146 carta - magna - 1215 - king - charter 5 146_carta_magna_1215_king
147 mccollum - reid - peru - cocaine - lima 5 147_mccollum_reid_peru_cocaine
148 film - scotland - location - outlander - movie 5 148_film_scotland_location_outlander

Training hyperparameters

  • calculate_probabilities: True
  • language: None
  • low_memory: False
  • min_topic_size: 10
  • n_gram_range: (1, 1)
  • nr_topics: 150
  • seed_topic_list: None
  • top_n_words: 10
  • verbose: False

Framework versions

  • Numpy: 1.23.5
  • HDBSCAN: 0.8.33
  • UMAP: 0.5.3
  • Pandas: 1.5.3
  • Scikit-Learn: 1.2.2
  • Sentence-transformers: 2.2.2
  • Transformers: 4.31.0
  • Numba: 0.57.1
  • Plotly: 5.15.0
  • Python: 3.10.12
Downloads last month
8
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.