predict_rugby

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("rociortizb/predict_rugby")

topic_model.get_topic_info()

Topic overview

• Number of topics: 40
• Number of training documents: 27774

Click here for an overview of all topics.

Topic ID	Topic Keywords	Topic Frequency	Label
0	blacks - new - new zealand - zealand - hurricanes	1556	0_blacks_new_new zealand_zealand
1	springboks - south - africa - south africa - erasmus	1297	1_springboks_south_africa_south africa
2	springbok - springboks - year - players - world	1205	2_springbok_springboks_year_players
3	stormers - lions - van - said - team	1163	3_stormers_lions_van_said
4	cheetahs - van - griquas - pumas - province	1149	4_cheetahs_van_griquas_pumas
5	sevens - series - fiji - blitzboks - pool	994	5_sevens_series_fiji_blitzboks
6	brumbies - reds - rebels - rugby - super	953	6_brumbies_reds_rebels_rugby
7	premiership - tom - exeter - wasps - saracens	928	7_premiership_tom_exeter_wasps
8	ireland - sexton - schmidt - irish - leinster	911	8_ireland_sexton_schmidt_irish
9	france - french - racing - year - club	904	9_france_french_racing_year
10	try - leinster - minutes - munster - penalty	876	10_try_leinster_minutes_munster
11	stormers - lions - south - game - team	870	11_stormers_lions_south_game
12	sharks - du - preez - du preez - bosch	851	12_sharks_du_preez_du preez
13	wallabies - australia - folau - rugby - said	837	13_wallabies_australia_folau_rugby
14	england - jones - harlequins - squad - george	807	14_england_jones_harlequins_squad
15	england - jones - world - world cup - wales	790	15_england_jones_world_world cup
16	crusaders - highlanders - hurricanes - blues - chiefs	769	16_crusaders_highlanders_hurricanes_blues
17	italy - france - england - ireland - scotland	763	17_italy_france_england_ireland
18	wallabies - australia - cheika - said - blacks	736	18_wallabies_australia_cheika_said
19	disciplinary - committee - foul play - foul - player	722	19_disciplinary_committee_foul play_foul
20	clermont - stade - montpellier - toulon - toulouse	688	20_clermont_stade_montpellier_toulon
21	blacks - new - zealand - new zealand - foster	673	21_blacks_new_zealand_new zealand
22	wales - davies - ospreys - scarlets - cardiff	666	22_wales_davies_ospreys_scarlets
23	bulls - van - stormers - lions - sharks	660	23_bulls_van_stormers_lions
24	bulls - van - white - rugby - loftus	615	24_bulls_van_white_rugby
25	rugby - super - super rugby - competition - new	531	25_rugby_super_super rugby_competition
26	scotland - glasgow - edinburgh - townsend - russell	529	26_scotland_glasgow_edinburgh_townsend
27	brumbies - waratahs - reds - rebels - force	521	27_brumbies_waratahs_reds_rebels
28	pro14 - leinster - ulster - scarlets - 19	490	28_pro14_leinster_ulster_scarlets
29	rugby - world - world rugby - nations - cup	467	29_rugby_world_world rugby_nations
30	argentina - santiago - pumas - juan - matias	447	30_argentina_santiago_pumas_juan
31	club - premiership - season - rugby - gloucester	436	31_club_premiership_season_rugby
32	club - premiership - saracens - wasps - salary	423	32_club_premiership_saracens_wasps
33	gatland - lions - wales - tour - barbarians	361	33_gatland_lions_wales_tour
34	africa - south africa - south - zealand - new zealand	339	34_africa_south africa_south_zealand
35	marais - saru - union - rugby - president	317	35_marais_saru_union_rugby
36	kings - southern kings - southern - davids - schalk	217	36_kings_southern kings_southern_davids
37	vs - referees - match official - official - assistant referees	171	37_vs_referees_match official_official
38	sunwolves - japan - super - super rugby - 15	116	38_sunwolves_japan_super_super rugby
39	burgess - lancaster - england - bath - union	26	39_burgess_lancaster_england_bath

Training hyperparameters

• calculate_probabilities: False
• language: None
• low_memory: False
• min_topic_size: 10
• n_gram_range: (1, 1)
• nr_topics: None
• seed_topic_list: [['United Rugby Championship', 'Ireland', 'Wales', 'Scotland', 'South Africa', 'Italian', 'Pro14'], ['Pro14', 'Edinburgh', 'Glasgow', 'Scarlets', 'Ospreys', 'Zebre', 'Benetton', 'Connacht', 'Leinster', 'Ulster', 'Munster'], ['European Cup', 'European', 'Heineken', 'competition', 'Toulon', 'Saracens', 'Leinster'], ['Premiership', 'England', 'Exeter', 'Saracens', 'Wasps', 'Leicester', 'Harlequins', 'Sale', 'Bristol', 'Northampton'], ['Sevens', 'Fiji', 'New Zealand', 'South Africa', 'England', 'Australia', 'series', 'HSBC', 'Olympics'], ['Super Rugby', 'New Zealand', 'Australia', 'South Africa', 'Argentina', 'Japan', 'Blues', 'Brumbies', 'Crusaders', 'Sharks', 'Stormers'], ['Six Nations', 'England', 'Wales', 'Ireland', 'Scotland', 'France', 'Italy', 'Championship', 'Grand Slam'], ['Currie Cup', 'South Africa', 'Bulls', 'Lions', 'Sharks', 'Cheetahs', 'Western Province', 'domestic', 'provincial'], ['World Cup', 'international', 'New Zealand', 'Australia', 'South Africa', 'England', 'Wales', 'France'], ['Rugby Championship', 'New Zealand', 'Australia', 'South Africa', 'Argentina', 'All Blacks', 'Wallabies', 'Springboks', 'Pumas'], ['British Irish Lions', 'South Africa', 'New Zealand', 'Australia']]
• top_n_words: 30
• verbose: True

Framework versions

• Numpy: 1.21.0
• HDBSCAN: 0.8.29
• UMAP: 0.5.3
• Pandas: 2.0.2
• Scikit-Learn: 1.2.2
• Sentence-transformers: 2.2.2
• Transformers: 4.29.2
• Numba: 0.57.0
• Plotly: 5.14.1
• Python: 3.9.6