jemole commited on
Commit
7cbbc68
·
verified ·
1 Parent(s): 28fc8c3

Reinicia los diccionarios con cada nuevo texto en entrenamiento

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -27,8 +27,17 @@ class NgramModel:
27
 
28
  def train(self, text: str):
29
  """Entrena el modelo creando diccionarios de n-gramas."""
 
 
 
 
 
 
 
 
 
30
  words = self.preprocess_text(text)
31
-
32
  # Crear n-gramas para n=1 hasta n=5
33
  for n in range(1, 6):
34
  for i in range(len(words) - n + 1):
 
27
 
28
  def train(self, text: str):
29
  """Entrena el modelo creando diccionarios de n-gramas."""
30
+ # Reiniciar los diccionarios
31
+ self.ngram_dicts = {
32
+ 1: defaultdict(int),
33
+ 2: defaultdict(int),
34
+ 3: defaultdict(int),
35
+ 4: defaultdict(int),
36
+ 5: defaultdict(int)
37
+ }
38
+
39
  words = self.preprocess_text(text)
40
+
41
  # Crear n-gramas para n=1 hasta n=5
42
  for n in range(1, 6):
43
  for i in range(len(words) - n + 1):