Add BERTopic model
Browse files- README.md +129 -0
- config.json +17 -0
- ctfidf.safetensors +3 -0
- ctfidf_config.json +0 -0
- topic_embeddings.safetensors +3 -0
- topics.json +0 -0
README.md
ADDED
|
@@ -0,0 +1,129 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
---
|
| 3 |
+
tags:
|
| 4 |
+
- bertopic
|
| 5 |
+
library_name: bertopic
|
| 6 |
+
pipeline_tag: text-classification
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# UPA1.0
|
| 10 |
+
|
| 11 |
+
This is a [BERTopic](https://github.com/MaartenGr/BERTopic) model.
|
| 12 |
+
BERTopic is a flexible and modular topic modeling framework that allows for the generation of easily interpretable topics from large datasets.
|
| 13 |
+
|
| 14 |
+
## Usage
|
| 15 |
+
|
| 16 |
+
To use this model, please install BERTopic:
|
| 17 |
+
|
| 18 |
+
```
|
| 19 |
+
pip install -U bertopic
|
| 20 |
+
```
|
| 21 |
+
|
| 22 |
+
You can use the model as follows:
|
| 23 |
+
|
| 24 |
+
```python
|
| 25 |
+
from bertopic import BERTopic
|
| 26 |
+
topic_model = BERTopic.load("JGamonalHML/UPA1.0")
|
| 27 |
+
|
| 28 |
+
topic_model.get_topic_info()
|
| 29 |
+
```
|
| 30 |
+
|
| 31 |
+
## Topic overview
|
| 32 |
+
|
| 33 |
+
* Number of topics: 60
|
| 34 |
+
* Number of training documents: 31789
|
| 35 |
+
|
| 36 |
+
<details>
|
| 37 |
+
<summary>Click here for an overview of all topics.</summary>
|
| 38 |
+
|
| 39 |
+
| Topic ID | Topic Keywords | Topic Frequency | Label |
|
| 40 |
+
|----------|----------------|-----------------|-------|
|
| 41 |
+
| -1 | sabroso - amability - ambiance - amias - arreglaron | 94 | -1_sabroso_amability_ambiance_amias |
|
| 42 |
+
| 0 | buena - atención - excelente - atencion - felicitaciones | 27 | 0_buena_atención_excelente_atencion |
|
| 43 |
+
| 1 | servicio - buen - bueno - excelente - amabilidad | 4761 | 1_servicio_buen_bueno_excelente |
|
| 44 |
+
| 2 | café - pan - cafe - sandwich - leche | 2756 | 2_café_pan_cafe_sandwich |
|
| 45 |
+
| 3 | personal - amable - lenta - atento - atiende | 1549 | 3_personal_amable_lenta_atento |
|
| 46 |
+
| 4 | cajera - cajero - caja - cajas - cajeras | 1673 | 4_cajera_cajero_caja_cajas |
|
| 47 |
+
| 5 | baño - baños - sucios - papel - sucio | 1521 | 5_baño_baños_sucios_papel |
|
| 48 |
+
| 6 | amabilidad - amables - amable - atención - personas | 1175 | 6_amabilidad_amables_amable_atención |
|
| 49 |
+
| 7 | productos - precios - precio - variedad - completos | 1151 | 7_productos_precios_precio_variedad |
|
| 50 |
+
| 8 | pago - efectivo - tarjeta - puntos - pantalla | 1212 | 8_pago_efectivo_tarjeta_puntos |
|
| 51 |
+
| 9 | rápida - rapidez - rápido - rapido - rapida | 1029 | 9_rápida_rapidez_rápido_rapido |
|
| 52 |
+
| 10 | cocina - chocolate - copec - caliente - poca | 862 | 10_cocina_chocolate_copec_caliente |
|
| 53 |
+
| 11 | chicas - niñas - señoritas - señorita - amables | 521 | 11_chicas_niñas_señoritas_señorita |
|
| 54 |
+
| 12 | turno - noche - tienda - tarde - nocturno | 545 | 12_turno_noche_tienda_tarde |
|
| 55 |
+
| 13 | limpieza - limpio - limpia - ordenado - tienda | 680 | 13_limpieza_limpio_limpia_ordenado |
|
| 56 |
+
| 14 | vendedora - vendedor - vendedoras - vendedores - shell | 693 | 14_vendedora_vendedor_vendedoras_vendedores |
|
| 57 |
+
| 15 | combustible - carga - bombero - auto - vehículo | 521 | 15_combustible_carga_bombero_auto |
|
| 58 |
+
| 16 | trabajo - trabajadores - trabajadoras - trabajar - equipo | 861 | 16_trabajo_trabajadores_trabajadoras_trabajar |
|
| 59 |
+
| 17 | excelente - servicio - atención - atencion - constanza | 661 | 17_excelente_servicio_atención_atencion |
|
| 60 |
+
| 18 | minutos - 10 - espera - 20 - 30 | 778 | 18_minutos_10_espera_20 |
|
| 61 |
+
| 19 | hot - dog - hamburguesas - hamburguesa - dogs | 576 | 19_hot_dog_hamburguesas_hamburguesa |
|
| 62 |
+
| 20 | cliente - clientes - trato - comentario - atención | 382 | 20_cliente_clientes_trato_comentario |
|
| 63 |
+
| 21 | atencion - buena - buenisima - atención - exelente | 772 | 21_atencion_buena_buenisima_atención |
|
| 64 |
+
| 22 | mala - malo - actitud - malos - calidad | 816 | 22_mala_malo_actitud_malos |
|
| 65 |
+
| 23 | aire - acondicionado - agua - olor - temperatura | 495 | 23_aire_acondicionado_agua_olor |
|
| 66 |
+
| 24 | sabor - sonrisa - moscas - mesón - música | 446 | 24_sabor_sonrisa_moscas_mesón |
|
| 67 |
+
| 25 | mañana - lunas - medias - mañanas - azúcar | 285 | 25_mañana_lunas_medias_mañanas |
|
| 68 |
+
| 26 | upa - upita - market - moneda - equipo | 225 | 26_upa_upita_market_moneda |
|
| 69 |
+
| 27 | pedido - pedidos - perros - perro - entrega | 318 | 27_pedido_pedidos_perros_perro |
|
| 70 |
+
| 28 | opciones - veganas - vegetarianas - vegetariana - vegetarianos | 245 | 28_opciones_veganas_vegetarianas_vegetariana |
|
| 71 |
+
| 29 | cordial - cordialidad - cordiales - amable - atención | 298 | 29_cordial_cordialidad_cordiales_amable |
|
| 72 |
+
| 30 | rica - comida - rico - fresca - café | 363 | 30_rica_comida_rico_fresca |
|
| 73 |
+
| 31 | linda - atención - hora - precisa - atencion | 179 | 31_linda_atención_hora_precisa |
|
| 74 |
+
| 32 | papas - fritas - empanadas - porción - aceite | 262 | 32_papas_fritas_empanadas_porción |
|
| 75 |
+
| 33 | bomberos - bombero - propina - combustible - isla | 266 | 33_bomberos_bombero_propina_combustible |
|
| 76 |
+
| 34 | comida - deliciosa - calidad - fresca - servicio | 190 | 34_comida_deliciosa_calidad_fresca |
|
| 77 |
+
| 35 | deficiente - educado - educada - mal - educación | 379 | 35_deficiente_educado_educada_mal |
|
| 78 |
+
| 36 | gentil - gabriela - gentiles - genial - gentileza | 244 | 36_gentil_gabriela_gentiles_genial |
|
| 79 |
+
| 37 | promociones - conceptos - principales - propina - comentario | 130 | 37_promociones_conceptos_principales_propina |
|
| 80 |
+
| 38 | lugar - limpio - agradable - lindo - ordenado | 204 | 38_lugar_limpio_agradable_lindo |
|
| 81 |
+
| 39 | message - personal - - - | 278 | 39_message_personal__ |
|
| 82 |
+
| 40 | carolina - karina - sra - cristina - maría | 61 | 40_carolina_karina_sra_cristina |
|
| 83 |
+
| 41 | lento - lavado - fichas - aspirado - auto | 91 | 41_lento_lavado_fichas_aspirado |
|
| 84 |
+
| 42 | paola - pamela - juan - ximena - dedicación | 166 | 42_paola_pamela_juan_ximena |
|
| 85 |
+
| 43 | alegría - alejandra - amorosa - diana - alejandro | 71 | 43_alegría_alejandra_amorosa_diana |
|
| 86 |
+
| 44 | limpios - baños - lindos - limpiar - olor | 78 | 44_limpios_baños_lindos_limpiar |
|
| 87 |
+
| 45 | mascotas - entrada - ingreso - mascota - permitir | 155 | 45_mascotas_entrada_ingreso_mascota |
|
| 88 |
+
| 46 | cortesía - cortes - cortés - mochila - usan | 74 | 46_cortesía_cortes_cortés_mochila |
|
| 89 |
+
| 47 | demora - hambre - entrega - apariencia - pedido | 50 | 47_demora_hambre_entrega_apariencia |
|
| 90 |
+
| 48 | madrugada - 04 - 03 - horas - tapas | 122 | 48_madrugada_04_03_horas |
|
| 91 |
+
| 49 | gel - alcohol - dispensador - dispensadores - ketchup | 42 | 49_gel_alcohol_dispensador_dispensadores |
|
| 92 |
+
| 50 | rico - rapido - fresco - sandwich - chocolate | 38 | 50_rico_rapido_fresco_sandwich |
|
| 93 |
+
| 51 | gasolina - región - trabaja - diésel - estación | 69 | 51_gasolina_región_trabaja_diésel |
|
| 94 |
+
| 52 | daniela - felicitar - romero - felipe - realizada | 49 | 52_daniela_felicitar_romero_felipe |
|
| 95 |
+
| 53 | ignacio - don - ivan - gran - quilaleo | 26 | 53_ignacio_don_ivan_gran |
|
| 96 |
+
| 54 | angelica - maria - srta - jose - especialmente | 39 | 54_angelica_maria_srta_jose |
|
| 97 |
+
| 55 | seguridad - guardia - robo - estacionamiento - miguel | 60 | 55_seguridad_guardia_robo_estacionamiento |
|
| 98 |
+
| 56 | cecilia - macaya - satisfecho - proporcionaron - gema | 83 | 56_cecilia_macaya_satisfecho_proporcionaron |
|
| 99 |
+
| 57 | gusta - gustó - gustaría - gusto - aquí | 44 | 57_gusta_gustó_gustaría_gusto |
|
| 100 |
+
| 58 | mechada - italiana - italiano - ensalada - recibido | 48 | 58_mechada_italiana_italiano_ensalada |
|
| 101 |
+
|
| 102 |
+
</details>
|
| 103 |
+
|
| 104 |
+
## Training hyperparameters
|
| 105 |
+
|
| 106 |
+
* calculate_probabilities: False
|
| 107 |
+
* language: None
|
| 108 |
+
* low_memory: False
|
| 109 |
+
* min_topic_size: 10
|
| 110 |
+
* n_gram_range: (1, 1)
|
| 111 |
+
* nr_topics: 60
|
| 112 |
+
* seed_topic_list: None
|
| 113 |
+
* top_n_words: 10
|
| 114 |
+
* verbose: False
|
| 115 |
+
* zeroshot_min_similarity: 0.7
|
| 116 |
+
* zeroshot_topic_list: None
|
| 117 |
+
|
| 118 |
+
## Framework versions
|
| 119 |
+
|
| 120 |
+
* Numpy: 2.2.5
|
| 121 |
+
* HDBSCAN: 0.8.40
|
| 122 |
+
* UMAP: 0.5.7
|
| 123 |
+
* Pandas: 2.2.3
|
| 124 |
+
* Scikit-Learn: 1.6.1
|
| 125 |
+
* Sentence-transformers: 4.1.0
|
| 126 |
+
* Transformers: 4.51.3
|
| 127 |
+
* Numba: 0.61.2
|
| 128 |
+
* Plotly: 6.0.1
|
| 129 |
+
* Python: 3.12.1
|
config.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"calculate_probabilities": false,
|
| 3 |
+
"language": null,
|
| 4 |
+
"low_memory": false,
|
| 5 |
+
"min_topic_size": 10,
|
| 6 |
+
"n_gram_range": [
|
| 7 |
+
1,
|
| 8 |
+
1
|
| 9 |
+
],
|
| 10 |
+
"nr_topics": 60,
|
| 11 |
+
"seed_topic_list": null,
|
| 12 |
+
"top_n_words": 10,
|
| 13 |
+
"verbose": false,
|
| 14 |
+
"zeroshot_min_similarity": 0.7,
|
| 15 |
+
"zeroshot_topic_list": null,
|
| 16 |
+
"embedding_model": "mrm8488/modernbert-embed-base-ft-sts-spanish-matryoshka-768-64"
|
| 17 |
+
}
|
ctfidf.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:037dd27c7f3faffa106facb77f9726551bd093b0fa98dd12af564dfc420bd45d
|
| 3 |
+
size 753260
|
ctfidf_config.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
topic_embeddings.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a00762d74e357f055fd2d003c07d30ac127e5dc66f6bee0eaf7ea59e835d2924
|
| 3 |
+
size 184408
|
topics.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|