@epfml/discojs-node 2.1.2-p20240507140056.0 → 2.1.2-p20240515132210.0

package/README.md

@@ -0,0 +1,7 @@
+# @epfml/discojs-node
+
+Decentralized & federated privacy-preserving ML training in TypeScript.
+
+This is the node-specific library of the Disco.js project,
+offering some helpers.
+You probably want to also use [`discojs`](../discojs).

package/dist/models/model_loader.d.ts

@@ -0,0 +1,3 @@
+import { models } from '@epfml/discojs';
+export declare function saveModelToDisk(model: models.Model, modelFolder: string, modelFileName: string): Promise<void>;
+export declare function loadModelFromDisk(modelPath: string): Promise<models.Model>;

package/dist/models/model_loader.js

@@ -0,0 +1,16 @@
+import fs from 'node:fs/promises';
+import { serialization } from '@epfml/discojs';
+export async function saveModelToDisk(model, modelFolder, modelFileName) {
+    try {
+        await fs.access(modelFolder);
+    }
+    catch {
+        await fs.mkdir(modelFolder);
+    }
+    const encoded = await serialization.model.encode(model);
+    await fs.writeFile(`${modelFolder}/${modelFileName}`, encoded);
+}
+export async function loadModelFromDisk(modelPath) {
+    const content = await fs.readFile(modelPath);
+    return await serialization.model.decode(content);
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@epfml/discojs-node",
-  "version": "2.1.2-p20240507140056.0",
+  "version": "2.1.2-p20240515132210.0",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",