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

UOM // Fix an index-out-of-range issue.

This commit is contained in:
ShikiSuen 2023-08-28 01:38:32 +08:00
parent bddbca6413
commit eb52d2e1ac
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Packages/vChewing_LangModelAssembly/Sources/LangModelAssembly/SubLMs/lmUserOverride.swift
View File

@ -239,7 +239,7 @@ public extension vChewingLM.LMUserOverride {
// MARK: - Private Methods // MARK: - Private Methods
extension vChewingLM.LMUserOverride { extension vChewingLM.LMUserOverride {
private func doObservation( func doObservation(
key: String, candidate: String, timestamp: Double, forceHighScoreOverride: Bool, key: String, candidate: String, timestamp: Double, forceHighScoreOverride: Bool,
saveCallback: @escaping () -> Void saveCallback: @escaping () -> Void
) { ) {
@ -254,7 +254,7 @@ extension vChewingLM.LMUserOverride {
mutLRUList.insert(koPair, at: 0) mutLRUList.insert(koPair, at: 0)
if mutLRUList.count > mutCapacity { if mutLRUList.count > mutCapacity {
mutLRUMap.removeValue(forKey: mutLRUList[mutLRUList.endIndex].key) mutLRUMap.removeValue(forKey: mutLRUList[mutLRUList.endIndex - 1].key)
mutLRUList.removeLast() mutLRUList.removeLast()
} }
vCLog("UOM: Observation finished with new observation: \(key)") vCLog("UOM: Observation finished with new observation: \(key)")
@ -273,7 +273,7 @@ extension vChewingLM.LMUserOverride {
} }
} }
private func getSuggestion(key: String, timestamp: Double, headReading: String) -> Suggestion { func getSuggestion(key: String, timestamp: Double, headReading: String) -> Suggestion {
guard !key.isEmpty, let kvPair = mutLRUMap[key] else { return .init() } guard !key.isEmpty, let kvPair = mutLRUMap[key] else { return .init() }
let observation: Observation = kvPair.observation let observation: Observation = kvPair.observation
var candidates: [(String, Megrez.Unigram)] = .init() var candidates: [(String, Megrez.Unigram)] = .init()
@ -298,7 +298,7 @@ extension vChewingLM.LMUserOverride {
return .init(candidates: candidates, forceHighScoreOverride: forceHighScoreOverride) return .init(candidates: candidates, forceHighScoreOverride: forceHighScoreOverride)
} }
private func getScore( func getScore(
eventCount: Int, eventCount: Int,
totalCount: Int, totalCount: Int,
eventTimestamp: Double, eventTimestamp: Double,
@ -311,7 +311,7 @@ extension vChewingLM.LMUserOverride {
return prob * decay return prob * decay
} }
private static func isPunctuation(_ node: Megrez.Node) -> Bool { static func isPunctuation(_ node: Megrez.Node) -> Bool {
for key in node.keyArray { for key in node.keyArray {
guard let firstChar = key.first else { continue } guard let firstChar = key.first else { continue }
return String(firstChar) == "_" return String(firstChar) == "_"
@ -319,7 +319,7 @@ extension vChewingLM.LMUserOverride {
return false return false
} }
private static func formObservationKey( static func formObservationKey(
walkedNodes: [Megrez.Node], headIndex cursorIndex: Int, readingOnly: Bool = false walkedNodes: [Megrez.Node], headIndex cursorIndex: Int, readingOnly: Bool = false
) -> String { ) -> String {
// let whiteList = "你他妳她祢衪它牠再在" // let whiteList = "你他妳她祢衪它牠再在"