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

Megrez // Fix Unigram.consolidate(), letting it handle nerfed scores.

This commit is contained in:
ShikiSuen 2022-10-02 13:36:13 +08:00
parent c3d3d95a1f
commit af6c11beab
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Packages/vChewing_Megrez/Sources/Megrez/8_Unigram.swift
View File

@ -43,12 +43,11 @@ extension Megrez {
extension Array where Element == Megrez.Unigram { extension Array where Element == Megrez.Unigram {
/// 給定過濾清單，讓單元圖陣列自我過濾。 /// 給定過濾清單，讓單元圖陣列自我過濾。
/// 在此基礎之上，對於相同詞值的多個單元圖，僅保留權重最大者。
public mutating func consolidate(filter theFilter: Set<String> = .init()) { public mutating func consolidate(filter theFilter: Set<String> = .init()) {
var inserted: [String: Double] = [:] var inserted: [String: Double] = [:]
var insertedArray: [Megrez.Unigram] = [] var insertedArray: [Megrez.Unigram] = []
for neta in filter({ !theFilter.contains($0.value) }) { for neta in filter({ !theFilter.contains($0.value) }) {
if let existed = inserted[neta.value], existed >= neta.score { continue } if inserted.keys.contains(neta.value) { continue }
inserted[neta.value] = neta.score inserted[neta.value] = neta.score
insertedArray.append(neta) insertedArray.append(neta)
} }