From 28e5c854cd9629dfc1f77eba2b16eeac4c6c6b4f Mon Sep 17 00:00:00 2001 From: Mingchung Xia <mingchung.xia@gmail.com> Date: Tue, 13 Feb 2024 22:26:44 -0500 Subject: [PATCH] Added DataHandler boilerplate --- .../HNSWCorpusDataHandler.swift | 27 +++++++++++++++++++ .../2. Encoding/ContextFreeEncoder.swift | 2 +- 2 files changed, 28 insertions(+), 1 deletion(-) create mode 100644 Sources/SwiftNLP/1. Data Collection/HNSWCorpusDataHandler.swift diff --git a/Sources/SwiftNLP/1. Data Collection/HNSWCorpusDataHandler.swift b/Sources/SwiftNLP/1. Data Collection/HNSWCorpusDataHandler.swift new file mode 100644 index 00000000..7035dbf6 --- /dev/null +++ b/Sources/SwiftNLP/1. Data Collection/HNSWCorpusDataHandler.swift @@ -0,0 +1,27 @@ +// +// HNSWCorpusDataHandler.swift +// +// +// Created by Mingchung Xia on 2024-02-13. +// + +import Foundation + +final class HNSWCorpusDataHandler<Scalar: BinaryFloatingPoint & Codable> { + private var corpus: HNSWCorpus<Scalar> + private var mmapURL: URL? // set default URL + + init(corpus: HNSWCorpus<Scalar>) { + self.corpus = corpus + } +} + +extension HNSWCorpusDataHandler { + func saveMemoryMap() { + + } + + func loadMemoryMap() { + + } +} diff --git a/Sources/SwiftNLP/2. Encoding/ContextFreeEncoder.swift b/Sources/SwiftNLP/2. Encoding/ContextFreeEncoder.swift index 3a67bd62..295c144d 100644 --- a/Sources/SwiftNLP/2. Encoding/ContextFreeEncoder.swift +++ b/Sources/SwiftNLP/2. Encoding/ContextFreeEncoder.swift @@ -69,7 +69,7 @@ class ContextFreeEncoder<Scalar: BinaryFloatingPoint & Codable>: SNLPEncoder { return } dictionary = ContextFreeEncoder<Scalar>.readDictionaryFromFile(url) - hnsw = ContextFreeEncoder<Scalar>.readHNSWFromFile(url) +// hnsw = ContextFreeEncoder<Scalar>.readHNSWFromFile(url) } subscript(_ token: String) -> [Scalar] { -- GitLab