Skip to content
Snippets Groups Projects
Commit d9b86682 authored by Mingchung Xia's avatar Mingchung Xia
Browse files

(unused) decodable conformance

parent cdc9f669
No related branches found
No related tags found
1 merge request!13HNSW Implementation with Testcases
...@@ -25,7 +25,7 @@ import Foundation ...@@ -25,7 +25,7 @@ import Foundation
class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus { class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus {
private var _documentEncoder: ContextFreeEncoder<Scalar> internal var _documentEncoder: ContextFreeEncoder<Scalar>
var zeroes: [Scalar] var zeroes: [Scalar]
var count: Int { 0 } var count: Int { 0 }
...@@ -38,9 +38,22 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus { ...@@ -38,9 +38,22 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus {
encodedDocuments = DeterministicSampleVectorIndex<[Scalar]>(typicalNeighborhoodSize: typicalNeighborhoodSize) encodedDocuments = DeterministicSampleVectorIndex<[Scalar]>(typicalNeighborhoodSize: typicalNeighborhoodSize)
} }
// Decodable conformance
// required init(from decoder: Decoder) throws {
// let container = try decoder.container(keyedBy: CodingKeys.self)
// _documentEncoder = try container.decode(ContextFreeEncoder<Scalar>.self, forKey: ._documentEncoder)
// zeroes = try container.decode([Scalar].self, forKey: .zeroes)
// encodedDocuments = try container.decode(DeterministicSampleVectorIndex<[Scalar]>.self, forKey: .encodedDocuments)
// }
@inlinable @inlinable
func addUntokenizedDocument(_ document: String) { func addUntokenizedDocument(_ document: String) {
encodedDocuments.insert((_documentEncoder.encodeSentence(document))) /// as! [Scalar] not needed encodedDocuments.insert((_documentEncoder.encodeSentence(document))) /// as! [Scalar] not needed
} }
// @inlinable
// func addVector(_ vector: [Scalar]) {
// encodedDocuments.insert(vector)
// }
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment