@epfml/discojs-node 2.0.0

package/dist/core/weights/index.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.aggregation = exports.WeightsContainer = void 0;
+var tslib_1 = require("tslib");
+var weights_container_1 = require("./weights_container");
+Object.defineProperty(exports, "WeightsContainer", { enumerable: true, get: function () { return weights_container_1.WeightsContainer; } });
+exports.aggregation = (0, tslib_1.__importStar)(require("./aggregation"));

package/dist/core/weights/weights_container.d.ts

@@ -0,0 +1,19 @@
+import { tf, Weights } from '..';
+export declare type TensorLike = tf.Tensor | ArrayLike<number>;
+export declare class WeightsContainer {
+    private readonly _weights;
+    constructor(weights: Iterable<TensorLike>);
+    get weights(): Weights;
+    add(other: WeightsContainer): WeightsContainer;
+    sub(other: WeightsContainer): WeightsContainer;
+    mapWith(other: WeightsContainer, fn: (a: tf.Tensor, b: tf.Tensor) => tf.Tensor): WeightsContainer;
+    map(fn: (t: tf.Tensor, i: number) => tf.Tensor): WeightsContainer;
+    map(fn: (t: tf.Tensor) => tf.Tensor): WeightsContainer;
+    reduce(fn: (acc: tf.Tensor, t: tf.Tensor) => tf.Tensor): tf.Tensor;
+    get(index: number): tf.Tensor | undefined;
+    frobeniusNorm(): number;
+    static of(...weights: TensorLike[]): WeightsContainer;
+    static from(model: tf.LayersModel): WeightsContainer;
+    static add(a: Iterable<TensorLike>, b: Iterable<TensorLike>): WeightsContainer;
+    static sub(a: Iterable<TensorLike>, b: Iterable<TensorLike>): WeightsContainer;
+}

package/dist/core/weights/weights_container.js

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WeightsContainer = void 0;
+var tslib_1 = require("tslib");
+var immutable_1 = require("immutable");
+var __1 = require("..");
+var WeightsContainer = /** @class */ (function () {
+    function WeightsContainer(weights) {
+        this._weights = (0, immutable_1.List)(weights).map(function (w) {
+            return w instanceof __1.tf.Tensor ? w : __1.tf.tensor(w);
+        });
+    }
+    Object.defineProperty(WeightsContainer.prototype, "weights", {
+        get: function () {
+            return this._weights.toArray();
+        },
+        enumerable: false,
+        configurable: true
+    });
+    WeightsContainer.prototype.add = function (other) {
+        return this.mapWith(other, __1.tf.add);
+    };
+    WeightsContainer.prototype.sub = function (other) {
+        return this.mapWith(other, __1.tf.sub);
+    };
+    WeightsContainer.prototype.mapWith = function (other, fn) {
+        return new WeightsContainer(this._weights
+            .zip(other._weights)
+            .map(function (_a) {
+            var _b = (0, tslib_1.__read)(_a, 2), w1 = _b[0], w2 = _b[1];
+            return fn(w1, w2);
+        }));
+    };
+    WeightsContainer.prototype.map = function (fn) {
+        return new WeightsContainer(this._weights.map(fn));
+    };
+    WeightsContainer.prototype.reduce = function (fn) {
+        return this._weights.reduce(fn);
+    };
+    WeightsContainer.prototype.get = function (index) {
+        return this._weights.get(index);
+    };
+    WeightsContainer.prototype.frobeniusNorm = function () {
+        return Math.sqrt(this.map(function (w) { return w.square().sum(); }).reduce(function (a, b) { return a.add(b); }).dataSync()[0]);
+    };
+    WeightsContainer.of = function () {
+        var weights = [];
+        for (var _i = 0; _i < arguments.length; _i++) {
+            weights[_i] = arguments[_i];
+        }
+        return new this(weights);
+    };
+    WeightsContainer.from = function (model) {
+        return new this(model.weights.map(function (w) { return w.read(); }));
+    };
+    WeightsContainer.add = function (a, b) {
+        return new this(a).add(new this(b));
+    };
+    WeightsContainer.sub = function (a, b) {
+        return new this(a).sub(new this(b));
+    };
+    return WeightsContainer;
+}());
+exports.WeightsContainer = WeightsContainer;

package/dist/dataset/data_loader/image_loader.d.ts

@@ -0,0 +1,4 @@
+import { tf, data } from '../..';
+export declare class NodeImageLoader extends data.ImageLoader<string> {
+    readImageFrom(source: string): Promise<tf.Tensor3D>;
+}

package/dist/dataset/data_loader/image_loader.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NodeImageLoader = void 0;
+var tslib_1 = require("tslib");
+var fs_1 = (0, tslib_1.__importDefault)(require("fs"));
+var __1 = require("../..");
+var NodeImageLoader = /** @class */ (function (_super) {
+    (0, tslib_1.__extends)(NodeImageLoader, _super);
+    function NodeImageLoader() {
+        return _super !== null && _super.apply(this, arguments) || this;
+    }
+    NodeImageLoader.prototype.readImageFrom = function (source) {
+        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
+            return (0, tslib_1.__generator)(this, function (_a) {
+                return [2 /*return*/, __1.tf.node.decodeImage(fs_1.default.readFileSync(source))];
+            });
+        });
+    };
+    return NodeImageLoader;
+}(__1.data.ImageLoader));
+exports.NodeImageLoader = NodeImageLoader;

package/dist/dataset/data_loader/index.d.ts

@@ -0,0 +1,2 @@
+export { NodeImageLoader } from './image_loader';
+export { NodeTabularLoader } from './tabular_loader';

package/dist/dataset/data_loader/index.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NodeTabularLoader = exports.NodeImageLoader = void 0;
+var image_loader_1 = require("./image_loader");
+Object.defineProperty(exports, "NodeImageLoader", { enumerable: true, get: function () { return image_loader_1.NodeImageLoader; } });
+var tabular_loader_1 = require("./tabular_loader");
+Object.defineProperty(exports, "NodeTabularLoader", { enumerable: true, get: function () { return tabular_loader_1.NodeTabularLoader; } });

package/dist/dataset/data_loader/tabular_loader.d.ts

@@ -0,0 +1,4 @@
+import { tf, data } from '../..';
+export declare class NodeTabularLoader extends data.TabularLoader<string> {
+    loadTabularDatasetFrom(source: string, csvConfig: Record<string, unknown>): tf.data.CSVDataset;
+}

package/dist/dataset/data_loader/tabular_loader.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NodeTabularLoader = void 0;
+var tslib_1 = require("tslib");
+var __1 = require("../..");
+var NodeTabularLoader = /** @class */ (function (_super) {
+    (0, tslib_1.__extends)(NodeTabularLoader, _super);
+    function NodeTabularLoader() {
+        return _super !== null && _super.apply(this, arguments) || this;
+    }
+    NodeTabularLoader.prototype.loadTabularDatasetFrom = function (source, csvConfig) {
+        var prefix = 'file://';
+        if (source.slice(0, 7) !== prefix) {
+            source = prefix + source;
+        }
+        return __1.tf.data.csv(source, csvConfig);
+    };
+    return NodeTabularLoader;
+}(__1.data.TabularLoader));
+exports.NodeTabularLoader = NodeTabularLoader;

package/dist/imports.d.ts

@@ -0,0 +1 @@
+export * as data from './dataset/data_loader';

package/dist/imports.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.data = void 0;
+var tslib_1 = require("tslib");
+exports.data = (0, tslib_1.__importStar)(require("./dataset/data_loader"));

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export * from './core';
+export * as node from './imports';

package/dist/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.node = void 0;
+var tslib_1 = require("tslib");
+(0, tslib_1.__exportStar)(require("./core"), exports);
+exports.node = (0, tslib_1.__importStar)(require("./imports"));

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "@epfml/discojs-node",
+  "version": "2.0.0",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "build": "cp ./src/core/index.node.ts ./src/core/index.ts && tsc",
+    "build-win": "copy ..\\discojs-core\\src\\index.node.ts ..\\discojs-core\\src\\index.ts && tsc",
+    "test": "cp ./src/core/index.node.ts ./src/core/index.ts && mocha",
+    "lint": "cp ./src/core/index.node.ts ./src/core/index.ts && npx eslint --max-warnings 0 --ignore-pattern '*.spec.ts' --ignore-pattern 'src/core' ."
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/epfml/disco.git"
+  },
+  "bugs": {
+    "url": "https://github.com/epfml/disco/issues"
+  },
+  "homepage": "https://github.com/epfml/disco#readme",
+  "dependencies": {
+    "msgpack-lite": "0.1",
+    "immutable": "4",
+    "tslib": "2",
+    "@tensorflow/tfjs-node": "4",
+    "isomorphic-ws": "4",
+    "url": "0.11",
+    "@koush/wrtc": "0.5",
+    "axios": "0.27",
+    "@types/msgpack-lite": "0.1",
+    "uuid": "8",
+    "ws": "8",
+    "simple-peer": "9"
+  }
+}