swift - 如何在 Swift(CreateML) 中读取文本文件的内容并将其解析为字典

Question

如何从文件中读取字典？例如我的文本文件如下所示：

[“book": [-0.33568978, -0.42831486],...]

我想读它并放入字典

let myVector = [String:[Double]]()

Answer 1

对于尝试使用 CreateML 导入自己的自定义嵌入的其他人，这是我的解决方案。

import Foundation
import PlaygroundSupport
import CreateML

do {
    let fileUrl = PlaygroundSupport.playgroundSharedDataDirectory.appendingPathComponent("resultsmall.json")
    let data = try Data(contentsOf: fileUrl)
    let decoder = JSONDecoder()
    let json = try decoder.decode([String: [Double]].self, from: data)

    let embedding = try MLWordEmbedding(dictionary: json)
    try embedding.write(to: URL(fileURLWithPath: "~/Desktop/WordEmbedding.mlmodel"))

} catch {
    print(error)
}