From b530eb8a5b75213efa7388591af7224631bd8e09 Mon Sep 17 00:00:00 2001
From: Mingchung Xia <mingchung.xia@gmail.com>
Date: Wed, 7 Feb 2024 23:36:19 -0500
Subject: [PATCH] Moved zeroes to computed field
---
Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift b/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift
index cd7c7272..da0cf183 100644
--- a/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift
+++ b/Sources/SwiftNLP/1. Data Collection/HNSWCorpus.swift
@@ -27,7 +27,7 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus {
// internal var _documentEncoder: ContextFreeEncoder<Scalar>
internal var _documentEncoder: any SNLPEncoder
- var zeroes: [Scalar]
+ var zeroes: [Scalar] { _documentEncoder.zeroes as! [Scalar] }
var encodedDocuments: DeterministicSampleVectorIndex<[Scalar]>
var count: Int { encodedDocuments.base.vectors.count }
@@ -36,13 +36,11 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus {
init(encoding: ContextFreeEncoder<Scalar>.PreComputedEmbeddings, scalar: Scalar.Type = Double.self,
typicalNeighborhoodSize: Int = 20) {
_documentEncoder = ContextFreeEncoder(source: encoding)
- zeroes = _documentEncoder.zeroes as! [Scalar]
encodedDocuments = DeterministicSampleVectorIndex<[Scalar]>(typicalNeighborhoodSize: typicalNeighborhoodSize)
}
init(encoder: any SNLPEncoder, scalar: Scalar.Type = Double.self, typicalNeighborhoodSize: Int = 20) {
_documentEncoder = encoder
- zeroes = _documentEncoder.zeroes as! [Scalar]
encodedDocuments = DeterministicSampleVectorIndex<[Scalar]>(typicalNeighborhoodSize: typicalNeighborhoodSize)
}
--
GitLab