refactor: format

thde · Feb 23, 2024 · da7c828 · da7c828
1 parent 6a2eda6
commit da7c828
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dictionary.go b/dictionary.go
@@ -88,7 +88,7 @@ func (w *Dictionary) count(substr string) {
 		w.counter[prefix] = map[string]int{}
 	}
 
-	w.counter[prefix][suffix] += 1
+	w.counter[prefix][suffix]++
 }
 
 // splitToLength splits a string to substrings of length.

diff --git a/generator.go b/generator.go
@@ -50,16 +50,18 @@ func (g Generator) Word() string {
 		if len(characters) > g.MaxSequences {
 			characters = characters[1:]
 		}
+
 		var nextAccumedProbs map[string]float32
 		n := 0
 		for {
 			str := strings.Join(characters[n:], "")
 			nextAccumedProbs = g.Probabilities[str]
-			n += 1
+			n++
 			if nextAccumedProbs != nil || n >= len(characters) {
 				break
 			}
 		}
+
 		nextCharacter := ""
 		r := random.Float32()
 		probability := float32(0)