floraachy
/
vChewing-macOS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

LMLite // Ask for score on query.

This commit is contained in:
ShikiSuen 2022-05-02 11:24:41 +08:00
parent 28889dc717
commit ec361b1893
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Source/Modules/LangModelRelated/SubLMs/lmLite.swift
View File

@ -29,14 +29,12 @@ import Foundation
extension vChewing { extension vChewing {
public class LMLite { public class LMLite {
var keyValueMap: [String: [Megrez.KeyValuePair]] = [:] var keyValueMap: [String: [Megrez.KeyValuePair]] = [:]
var defaultScore: Double = 0
var theData: String = "" var theData: String = ""
var allowConsolidation = false var allowConsolidation = false
public init(defaultScore scoreDefault: Double = 0, consolidate: Bool = false) { public init(consolidate: Bool = false) {
keyValueMap = [:] keyValueMap = [:]
theData = "" theData = ""
defaultScore = scoreDefault
allowConsolidation = consolidate allowConsolidation = consolidate
} }
@ -125,11 +123,11 @@ extension vChewing {
IME.prtDebugIntel(strDump) IME.prtDebugIntel(strDump)
} }
public func unigramsFor(key: String) -> [Megrez.Unigram] { public func unigramsFor(key: String, score givenScore: Double = 0.0) -> [Megrez.Unigram] {
var v: [Megrez.Unigram] = [] var v: [Megrez.Unigram] = []
if let matched = keyValueMap[key] { if let matched = keyValueMap[key] {
for entry in matched as [Megrez.KeyValuePair] { for entry in matched as [Megrez.KeyValuePair] {
v.append(Megrez.Unigram(keyValue: entry, score: defaultScore)) v.append(Megrez.Unigram(keyValue: entry, score: givenScore))
} }
} }
return v return v