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

Added documentation to HNSWCorpus

parent f4b01ad6
No related branches found
No related tags found
1 merge request!13HNSW Implementation with Testcases
Pipeline #110123 passed with warnings
...@@ -27,7 +27,7 @@ import PriorityHeapAlgorithms ...@@ -27,7 +27,7 @@ import PriorityHeapAlgorithms
import SimilarityMetric import SimilarityMetric
import HNSWAlgorithm import HNSWAlgorithm
import HNSWEphemeral import HNSWEphemeral
import GameplayKit import GameplayKit /// Not avaliable on Linux?
class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus { class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus {
...@@ -46,7 +46,7 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus { ...@@ -46,7 +46,7 @@ class HNSWCorpus<Scalar: BinaryFloatingPoint & Codable>: SNLPCorpus {
@inlinable @inlinable
func addUntokenizedDocument(_ document: String) { func addUntokenizedDocument(_ document: String) {
// Forced cast to [Scalar] is unnecessary /// Forced cast to [Scalar] is unnecessary
encodedDocuments.insert((_documentEncoder.encodeSentence(document))) encodedDocuments.insert((_documentEncoder.encodeSentence(document)))
} }
} }
...@@ -78,6 +78,7 @@ public struct DeterministicSampleVectorIndex<Vector: Collection & Codable> where ...@@ -78,6 +78,7 @@ public struct DeterministicSampleVectorIndex<Vector: Collection & Codable> where
public mutating func insert(_ vector: Vector) { public mutating func insert(_ vector: Vector) {
let convertedVector: [Double] = vector.map{ Double($0) } let convertedVector: [Double] = vector.map{ Double($0) }
if let metricVector = convertedVector as? CartesianDistanceMetric<Vector>.Vector { if let metricVector = convertedVector as? CartesianDistanceMetric<Vector>.Vector {
/// Returns a Key (unused)
base.insert(metricVector, using: &graphRNG) base.insert(metricVector, using: &graphRNG)
} else { } else {
fatalError("Unable to get metric vector") fatalError("Unable to get metric 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