kiss-lang/kiss-vscode
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

handloose fiddle with model frequencies

Browse Source
This commit is contained in:
Nat Quayle Nelson
2022-01-28 14:01:57 -07:00
parent 4933f3d3d2
commit 8ded66c2d7
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
projects/handloose/source/DocumentModel.kiss
View File

@@ -12,15 +12,17 @@
(File.saveContent path content)
(print content))
(prop :Map<String,Map<String,Int>> charFrequencies (new Map))
(prop :Map<String,Map<String,Int>> wordFrequencies (new Map))
(prop :Map<String,Map<String,Float>> charFrequencies (new Map))
(prop :Map<String,Map<String,Float>> wordFrequencies (new Map))
(method incFrequency [:Map<String,Map<String,Int>> m :String c :String following]
(when c
(when (= c "\r") (set c "\n"))
(unless (m.exists following) (dictSet m following (new Map)))
(let [followingMap (dictGet m following)]
(dictSet followingMap c (+ 1 (or (dictGet followingMap c) 0))))))
(method incFrequency [:Map<String,Map<String,Float>> m :String c :String following]
(let [&mut weight 1.0]
(when c
(when (= c "\r") (set c "\n"))
(when (= c following " ") (set c "\t") (set weight 0.25))
(unless (m.exists following) (dictSet m following (new Map)))
(let [followingMap (dictGet m following)]
(dictSet followingMap c (+ weight (or (dictGet followingMap c) 0)))))))
(method learnFrequencies [:String str]
(let [chars (str.split "")]
@@ -32,6 +34,7 @@
))
(prop :FlxRandom r (new FlxRandom))
(var ANY_CHANCE 25) // percent
(method :ArrowStuff generateArrowStuff []
// TODO also generate word arrows if lastChar is a space
(let [lastChar (substr content -1)
@@ -41,5 +44,5 @@
(doFor =>c weight charFreq
(chars.push c)
(weights.push weight))
(let [c (r.getObject chars weights)]
(let [c (r.getObject chars (if (r.bool ANY_CHANCE) null weights))]
(object text c action ->(type c)))))