@epfml/discojs-node 2.1.1 → 2.1.2-p20240506085559.0

package/dist/core/informant/training_informant/base.js

@@ -1,83 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Base = void 0;
-var immutable_1 = require("immutable");
-var graph_informant_1 = require("../graph_informant");
-var Base = /** @class */ (function () {
-    function Base(task, nbrMessagesToShow) {
-        if (nbrMessagesToShow === void 0) { nbrMessagesToShow = 10; }
-        this.task = task;
-        this.nbrMessagesToShow = nbrMessagesToShow;
-        // written feedback
-        this.messages = (0, immutable_1.List)();
-        // graph feedback
-        this.trainingGraphInformant = new graph_informant_1.GraphInformant();
-        this.validationGraphInformant = new graph_informant_1.GraphInformant();
-        // statistics
-        this.currentRound = 0;
-        this.currentNumberOfParticipants = 0;
-        this.totalNumberOfParticipants = 0;
-        this.averageNumberOfParticipants = 0;
-    }
-    Base.prototype.addMessage = function (msg) {
-        if (this.messages.size >= this.nbrMessagesToShow) {
-            this.messages = this.messages.shift();
-        }
-        this.messages = this.messages.push(msg);
-    };
-    Base.prototype.getMessages = function () {
-        return this.messages.toArray();
-    };
-    Base.prototype.round = function () {
-        return this.currentRound;
-    };
-    Base.prototype.participants = function () {
-        return this.currentNumberOfParticipants;
-    };
-    Base.prototype.totalParticipants = function () {
-        return this.totalNumberOfParticipants;
-    };
-    Base.prototype.averageParticipants = function () {
-        return this.averageNumberOfParticipants;
-    };
-    Base.prototype.updateTrainingGraph = function (accuracy) {
-        this.trainingGraphInformant.updateAccuracy(accuracy);
-    };
-    Base.prototype.updateValidationGraph = function (accuracy) {
-        this.validationGraphInformant.updateAccuracy(accuracy);
-    };
-    Base.prototype.trainingAccuracy = function () {
-        return this.trainingGraphInformant.accuracy();
-    };
-    Base.prototype.validationAccuracy = function () {
-        return this.validationGraphInformant.accuracy();
-    };
-    Base.prototype.trainingAccuracyData = function () {
-        return this.trainingGraphInformant.data();
-    };
-    Base.prototype.validationAccuracyData = function () {
-        return this.validationGraphInformant.data();
-    };
-    Base.prototype.isDecentralized = function () {
-        return false;
-    };
-    Base.prototype.isFederated = function () {
-        return false;
-    };
-    Base.isTrainingInformant = function (raw) {
-        if (typeof raw !== 'object') {
-            return false;
-        }
-        if (raw === null) {
-            return false;
-        }
-        // TODO
-        var requiredFields = (0, immutable_1.Set)();
-        if (!(requiredFields.every(function (field) { return field in raw; }))) {
-            return false;
-        }
-        return true;
-    };
-    return Base;
-}());
-exports.Base = Base;

package/dist/core/informant/training_informant/decentralized.d.ts

@@ -1,5 +0,0 @@
-import { Base } from '.';
-export declare class DecentralizedInformant extends Base {
-    update(statistics: Record<string, number>): void;
-    isDecentralized(): boolean;
-}

package/dist/core/informant/training_informant/decentralized.js

@@ -1,22 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DecentralizedInformant = void 0;
-var tslib_1 = require("tslib");
-var _1 = require(".");
-var DecentralizedInformant = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(DecentralizedInformant, _super);
-    function DecentralizedInformant() {
-        return _super !== null && _super.apply(this, arguments) || this;
-    }
-    DecentralizedInformant.prototype.update = function (statistics) {
-        this.currentRound += 1;
-        this.currentNumberOfParticipants = statistics.currentNumberOfParticipants;
-        this.totalNumberOfParticipants += this.currentNumberOfParticipants;
-        this.averageNumberOfParticipants = this.totalNumberOfParticipants / this.currentRound;
-    };
-    DecentralizedInformant.prototype.isDecentralized = function () {
-        return true;
-    };
-    return DecentralizedInformant;
-}(_1.Base));
-exports.DecentralizedInformant = DecentralizedInformant;

package/dist/core/informant/training_informant/federated.d.ts

@@ -1,14 +0,0 @@
-import { Base } from '.';
-/**
- * Class that collects information about the status of the training-loop of the model.
- */
-export declare class FederatedInformant extends Base {
-    displayHeatmap: boolean;
-    /**
-     * Update the server statistics with the JSON received from the server
-     * For now it's just the JSON, but we might want to keep it as a dictionary
-     * @param receivedStatistics statistics received from the server.
-     */
-    update(receivedStatistics: Record<string, number>): void;
-    isFederated(): boolean;
-}

package/dist/core/informant/training_informant/federated.js

@@ -1,32 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.FederatedInformant = void 0;
-var tslib_1 = require("tslib");
-var _1 = require(".");
-/**
- * Class that collects information about the status of the training-loop of the model.
- */
-var FederatedInformant = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(FederatedInformant, _super);
-    function FederatedInformant() {
-        var _this = _super !== null && _super.apply(this, arguments) || this;
-        _this.displayHeatmap = false;
-        return _this;
-    }
-    /**
-     * Update the server statistics with the JSON received from the server
-     * For now it's just the JSON, but we might want to keep it as a dictionary
-     * @param receivedStatistics statistics received from the server.
-     */
-    FederatedInformant.prototype.update = function (receivedStatistics) {
-        this.currentRound = receivedStatistics.round;
-        this.currentNumberOfParticipants = receivedStatistics.currentNumberOfParticipants;
-        this.totalNumberOfParticipants = receivedStatistics.totalNumberOfParticipants;
-        this.averageNumberOfParticipants = receivedStatistics.averageNumberOfParticipants;
-    };
-    FederatedInformant.prototype.isFederated = function () {
-        return true;
-    };
-    return FederatedInformant;
-}(_1.Base));
-exports.FederatedInformant = FederatedInformant;

package/dist/core/informant/training_informant/index.d.ts

@@ -1,4 +0,0 @@
-export { Base } from './base';
-export { FederatedInformant } from './federated';
-export { DecentralizedInformant } from './decentralized';
-export { LocalInformant } from './local';

package/dist/core/informant/training_informant/index.js

@@ -1,11 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.LocalInformant = exports.DecentralizedInformant = exports.FederatedInformant = exports.Base = void 0;
-var base_1 = require("./base");
-Object.defineProperty(exports, "Base", { enumerable: true, get: function () { return base_1.Base; } });
-var federated_1 = require("./federated");
-Object.defineProperty(exports, "FederatedInformant", { enumerable: true, get: function () { return federated_1.FederatedInformant; } });
-var decentralized_1 = require("./decentralized");
-Object.defineProperty(exports, "DecentralizedInformant", { enumerable: true, get: function () { return decentralized_1.DecentralizedInformant; } });
-var local_1 = require("./local");
-Object.defineProperty(exports, "LocalInformant", { enumerable: true, get: function () { return local_1.LocalInformant; } });

package/dist/core/informant/training_informant/local.d.ts

@@ -1,6 +0,0 @@
-import { Task } from '../../task';
-import { Base } from '.';
-export declare class LocalInformant extends Base {
-    constructor(task: Task, nbrMessagesToShow?: number);
-    update(statistics: Record<string, number>): void;
-}

package/dist/core/informant/training_informant/local.js

@@ -1,20 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.LocalInformant = void 0;
-var tslib_1 = require("tslib");
-var _1 = require(".");
-var LocalInformant = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(LocalInformant, _super);
-    function LocalInformant(task, nbrMessagesToShow) {
-        var _this = _super.call(this, task, nbrMessagesToShow) || this;
-        _this.currentNumberOfParticipants = 1;
-        _this.averageNumberOfParticipants = 1;
-        _this.totalNumberOfParticipants = 1;
-        return _this;
-    }
-    LocalInformant.prototype.update = function (statistics) {
-        this.currentRound = statistics.currentRound;
-    };
-    return LocalInformant;
-}(_1.Base));
-exports.LocalInformant = LocalInformant;

package/dist/core/logging/console_logger.d.ts

@@ -1,18 +0,0 @@
-import { Logger } from './logger';
-/**
- * Same properties as Toaster but on the console
- *
- * @class Logger
- */
-export declare class ConsoleLogger extends Logger {
-    /**
-     * Logs success message on the console (in green)
-     * @param {String} message - message to be displayed
-     */
-    success(message: string): void;
-    /**
-     * Logs error message on the console (in red)
-     * @param message - message to be displayed
-     */
-    error(message: string): void;
-}

package/dist/core/logging/console_logger.js

@@ -1,33 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ConsoleLogger = void 0;
-var tslib_1 = require("tslib");
-var chalk_1 = (0, tslib_1.__importDefault)(require("chalk"));
-var logger_1 = require("./logger");
-/**
- * Same properties as Toaster but on the console
- *
- * @class Logger
- */
-var ConsoleLogger = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(ConsoleLogger, _super);
-    function ConsoleLogger() {
-        return _super !== null && _super.apply(this, arguments) || this;
-    }
-    /**
-     * Logs success message on the console (in green)
-     * @param {String} message - message to be displayed
-     */
-    ConsoleLogger.prototype.success = function (message) {
-        console.log(chalk_1.default.green(message));
-    };
-    /**
-     * Logs error message on the console (in red)
-     * @param message - message to be displayed
-     */
-    ConsoleLogger.prototype.error = function (message) {
-        console.log(chalk_1.default.red(message));
-    };
-    return ConsoleLogger;
-}(logger_1.Logger));
-exports.ConsoleLogger = ConsoleLogger;

package/dist/core/logging/index.d.ts

@@ -1,3 +0,0 @@
-export { Logger } from './logger';
-export { ConsoleLogger } from './console_logger';
-export { TrainerLog } from './trainer_logger';

package/dist/core/logging/index.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TrainerLog = exports.ConsoleLogger = exports.Logger = void 0;
-var logger_1 = require("./logger");
-Object.defineProperty(exports, "Logger", { enumerable: true, get: function () { return logger_1.Logger; } });
-var console_logger_1 = require("./console_logger");
-Object.defineProperty(exports, "ConsoleLogger", { enumerable: true, get: function () { return console_logger_1.ConsoleLogger; } });
-var trainer_logger_1 = require("./trainer_logger");
-Object.defineProperty(exports, "TrainerLog", { enumerable: true, get: function () { return trainer_logger_1.TrainerLog; } });

package/dist/core/logging/logger.d.ts

@@ -1,12 +0,0 @@
-export declare abstract class Logger {
-    /**
-     * Logs sucess message (in green)
-     * @param message - message to be displayed
-     */
-    abstract success(message: string): void;
-    /**
-     * Logs error message (in red)
-     * @param message - message to be displayed
-     */
-    abstract error(message: string): void;
-}

package/dist/core/logging/logger.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Logger = void 0;
-var Logger = /** @class */ (function () {
-    function Logger() {
-    }
-    return Logger;
-}());
-exports.Logger = Logger;

package/dist/core/logging/trainer_logger.d.ts

@@ -1,24 +0,0 @@
-import { List } from 'immutable';
-import { tf } from '..';
-import { ConsoleLogger } from '.';
-export declare class TrainerLog {
-    epochs: List<number>;
-    trainAccuracy: List<number>;
-    validationAccuracy: List<number>;
-    loss: List<number>;
-    add(epoch: number, logs?: tf.Logs): void;
-}
-/**
- *
- * @class TrainerLogger
- */
-export declare class TrainerLogger extends ConsoleLogger {
-    readonly log: TrainerLog;
-    readonly saveTrainerLog: boolean;
-    constructor(saveTrainerLog?: boolean);
-    onEpochEnd(epoch: number, logs?: tf.Logs): void;
-    /**
-     *  Display ram usage
-     */
-    ramUsage(): void;
-}

package/dist/core/logging/trainer_logger.js

@@ -1,59 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TrainerLogger = exports.TrainerLog = void 0;
-var tslib_1 = require("tslib");
-var immutable_1 = require("immutable");
-var __1 = require("..");
-var _1 = require(".");
-var TrainerLog = /** @class */ (function () {
-    function TrainerLog() {
-        this.epochs = (0, immutable_1.List)();
-        this.trainAccuracy = (0, immutable_1.List)();
-        this.validationAccuracy = (0, immutable_1.List)();
-        this.loss = (0, immutable_1.List)();
-    }
-    TrainerLog.prototype.add = function (epoch, logs) {
-        this.epochs = this.epochs.push(epoch);
-        if (logs !== undefined) {
-            this.trainAccuracy = this.trainAccuracy.push(logs.acc);
-            this.validationAccuracy = this.validationAccuracy.push(logs.val_acc);
-            this.loss = this.loss.push(logs.loss);
-        }
-    };
-    return TrainerLog;
-}());
-exports.TrainerLog = TrainerLog;
-/**
- *
- * @class TrainerLogger
- */
-var TrainerLogger = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(TrainerLogger, _super);
-    // TODO: pass savaTrainerLog as false in browser, used for benchmarking
-    function TrainerLogger(saveTrainerLog) {
-        if (saveTrainerLog === void 0) { saveTrainerLog = true; }
-        var _this = _super.call(this) || this;
-        _this.saveTrainerLog = saveTrainerLog;
-        _this.log = new TrainerLog();
-        return _this;
-    }
-    TrainerLogger.prototype.onEpochEnd = function (epoch, logs) {
-        var _a, _b, _c;
-        // save logs
-        if (this.saveTrainerLog) {
-            this.log.add(epoch, logs);
-        }
-        // console output
-        var msg = "Epoch: " + epoch + "\nTrain: " + ((_a = logs === null || logs === void 0 ? void 0 : logs.acc) !== null && _a !== void 0 ? _a : 'undefined') + "\nValidation:" + ((_b = logs === null || logs === void 0 ? void 0 : logs.val_acc) !== null && _b !== void 0 ? _b : 'undefined') + "\nLoss:" + ((_c = logs === null || logs === void 0 ? void 0 : logs.loss) !== null && _c !== void 0 ? _c : 'undefined');
-        this.success("On epoch end:\n" + msg + "\n");
-    };
-    /**
-     *  Display ram usage
-     */
-    TrainerLogger.prototype.ramUsage = function () {
-        this.success("Training RAM usage is  = " + __1.tf.memory().numBytes * 0.000001 + " MB");
-        this.success("Number of allocated tensors  = " + __1.tf.memory().numTensors);
-    };
-    return TrainerLogger;
-}(_1.ConsoleLogger));
-exports.TrainerLogger = TrainerLogger;

package/dist/core/memory/base.d.ts

@@ -1,22 +0,0 @@
-import * as tf from '@tensorflow/tfjs';
-import { TaskID } from '..';
-import { ModelType } from './model_type';
-export declare type Path = string;
-export interface ModelInfo {
-    type?: ModelType;
-    taskID: TaskID;
-    name: string;
-}
-export declare type ModelSource = ModelInfo | Path;
-export declare abstract class Memory {
-    abstract getModel(source: ModelSource): Promise<tf.LayersModel>;
-    abstract deleteModel(source: ModelSource): Promise<void>;
-    abstract loadModel(source: ModelSource): Promise<void>;
-    abstract getModelMetadata(source: ModelSource): Promise<object | undefined>;
-    abstract updateWorkingModel(source: ModelSource, model: tf.LayersModel): Promise<void>;
-    abstract saveWorkingModel(source: ModelSource): Promise<void>;
-    abstract downloadModel(source: ModelSource): Promise<void>;
-    abstract contains(source: ModelSource): Promise<boolean>;
-    abstract pathFor(source: ModelSource): Path | undefined;
-    abstract infoFor(source: ModelSource): ModelInfo | undefined;
-}

package/dist/core/memory/base.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Memory = void 0;
-var Memory = /** @class */ (function () {
-    function Memory() {
-    }
-    return Memory;
-}());
-exports.Memory = Memory;

package/dist/core/memory/empty.d.ts

@@ -1,14 +0,0 @@
-import { tf } from '..';
-import { Memory, ModelInfo, Path } from './base';
-export declare class Empty extends Memory {
-    getModelMetadata(): Promise<undefined>;
-    contains(): Promise<boolean>;
-    getModel(): Promise<tf.LayersModel>;
-    loadModel(): Promise<void>;
-    updateWorkingModel(): Promise<void>;
-    saveWorkingModel(): Promise<void>;
-    deleteModel(): Promise<void>;
-    downloadModel(): Promise<void>;
-    pathFor(): Path;
-    infoFor(): ModelInfo;
-}

package/dist/core/memory/empty.js

@@ -1,75 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Empty = void 0;
-var tslib_1 = require("tslib");
-var base_1 = require("./base");
-var Empty = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(Empty, _super);
-    function Empty() {
-        return _super !== null && _super.apply(this, arguments) || this;
-    }
-    Empty.prototype.getModelMetadata = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                return [2 /*return*/, undefined];
-            });
-        });
-    };
-    Empty.prototype.contains = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                return [2 /*return*/, false];
-            });
-        });
-    };
-    Empty.prototype.getModel = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                throw new Error('empty');
-            });
-        });
-    };
-    Empty.prototype.loadModel = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                throw new Error('empty');
-            });
-        });
-    };
-    Empty.prototype.updateWorkingModel = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                return [2 /*return*/];
-            });
-        });
-    };
-    Empty.prototype.saveWorkingModel = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                return [2 /*return*/];
-            });
-        });
-    };
-    Empty.prototype.deleteModel = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                return [2 /*return*/];
-            });
-        });
-    };
-    Empty.prototype.downloadModel = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                throw new Error('empty');
-            });
-        });
-    };
-    Empty.prototype.pathFor = function () {
-        throw new Error('empty');
-    };
-    Empty.prototype.infoFor = function () {
-        throw new Error('empty');
-    };
-    return Empty;
-}(base_1.Memory));
-exports.Empty = Empty;

package/dist/core/memory/index.d.ts

@@ -1,3 +0,0 @@
-export { Empty } from './empty';
-export { Memory, ModelInfo, Path, ModelSource } from './base';
-export { ModelType } from './model_type';

package/dist/core/memory/index.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ModelType = exports.Memory = exports.Empty = void 0;
-var empty_1 = require("./empty");
-Object.defineProperty(exports, "Empty", { enumerable: true, get: function () { return empty_1.Empty; } });
-var base_1 = require("./base");
-Object.defineProperty(exports, "Memory", { enumerable: true, get: function () { return base_1.Memory; } });
-var model_type_1 = require("./model_type");
-Object.defineProperty(exports, "ModelType", { enumerable: true, get: function () { return model_type_1.ModelType; } });

package/dist/core/memory/model_type.d.ts

@@ -1,4 +0,0 @@
-export declare enum ModelType {
-    WORKING = "working",
-    SAVED = "saved"
-}

package/dist/core/memory/model_type.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ModelType = void 0;
-// Type of model to store & retrieve
-var ModelType;
-(function (ModelType) {
-    ModelType["WORKING"] = "working";
-    ModelType["SAVED"] = "saved";
-})(ModelType = exports.ModelType || (exports.ModelType = {}));

package/dist/core/privacy.d.ts

@@ -1,11 +0,0 @@
-import { Task, WeightsContainer } from '.';
-/**
- * Add task-parametrized Gaussian noise to and clip the weights update between the previous and current rounds.
- * The previous round's weights are the last weights pulled from server/peers.
- * The current round's weights are obtained after a single round of training, from the previous round's weights.
- * @param updatedWeights weights from the current round
- * @param staleWeights weights from the previous round
- * @param task the task
- * @returns the noised weights for the current round
- */
-export declare function addDifferentialPrivacy(updatedWeights: WeightsContainer, staleWeights: WeightsContainer, task: Task): WeightsContainer;

package/dist/core/privacy.js

@@ -1,47 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.addDifferentialPrivacy = void 0;
-var _1 = require(".");
-/**
- * Add task-parametrized Gaussian noise to and clip the weights update between the previous and current rounds.
- * The previous round's weights are the last weights pulled from server/peers.
- * The current round's weights are obtained after a single round of training, from the previous round's weights.
- * @param updatedWeights weights from the current round
- * @param staleWeights weights from the previous round
- * @param task the task
- * @returns the noised weights for the current round
- */
-function addDifferentialPrivacy(updatedWeights, staleWeights, task) {
-    var _a, _b;
-    var noiseScale = (_a = task.trainingInformation) === null || _a === void 0 ? void 0 : _a.noiseScale;
-    var clippingRadius = (_b = task.trainingInformation) === null || _b === void 0 ? void 0 : _b.clippingRadius;
-    var weightsDiff = updatedWeights.sub(staleWeights);
-    var newWeightsDiff;
-    if (clippingRadius !== undefined) {
-        // Frobenius norm
-        var norm_1 = weightsDiff.frobeniusNorm();
-        newWeightsDiff = weightsDiff.map(function (w) {
-            var clipped = w.div(Math.max(1, norm_1 / clippingRadius));
-            if (noiseScale !== undefined) {
-                // Add clipping and noise
-                var noise = _1.tf.randomNormal(w.shape, 0, (noiseScale * noiseScale) * (clippingRadius * clippingRadius));
-                return clipped.add(noise);
-            }
-            else {
-                // Add clipping without any noise
-                return clipped;
-            }
-        });
-    }
-    else {
-        if (noiseScale !== undefined) {
-            // Add noise without any clipping
-            newWeightsDiff = weightsDiff.map(function (w) { return _1.tf.randomNormal(w.shape, 0, (noiseScale * noiseScale)); });
-        }
-        else {
-            return updatedWeights;
-        }
-    }
-    return staleWeights.add(newWeightsDiff);
-}
-exports.addDifferentialPrivacy = addDifferentialPrivacy;

package/dist/core/serialization/index.d.ts

@@ -1,2 +0,0 @@
-export * as model from './model';
-export * as weights from './weights';

package/dist/core/serialization/index.js

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

package/dist/core/serialization/model.d.ts

@@ -1,5 +0,0 @@
-import { tf } from '..';
-export declare type Encoded = number[];
-export declare function isEncoded(raw: unknown): raw is Encoded;
-export declare function encode(model: tf.LayersModel): Promise<Encoded>;
-export declare function decode(encoded: Encoded): Promise<tf.LayersModel>;