@epfml/discojs 2.1.1 → 2.1.2-p20240506085037.0

package/dist/core/default_tasks/lus_covid.d.ts

@@ -1,2 +0,0 @@
-import { TaskProvider } from '..';
-export declare const lusCovid: TaskProvider;

package/dist/core/default_tasks/lus_covid.js

@@ -1,96 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.lusCovid = void 0;
-var tslib_1 = require("tslib");
-var __1 = require("..");
-exports.lusCovid = {
-    getTask: function () {
-        return {
-            taskID: 'lus_covid',
-            displayInformation: {
-                taskTitle: 'COVID Lung Ultrasound',
-                summary: {
-                    preview: 'Do you have a data of lung ultrasound images on patients <b>suspected of Lower Respiratory Tract infection (LRTI) during the COVID pandemic</b>? <br> Learn how to discriminate between COVID positive and negative patients by joining this task.',
-                    overview: "Don’t have a dataset of your own? Download a sample of a few cases <a class='underline' href='https://drive.switch.ch/index.php/s/zM5ZrUWK3taaIly' target='_blank'>here</a>."
-                },
-                model: "We use a simplified* version of the <b>DeepChest model</b>: A deep learning model developed in our lab (<a class='underline' href='https://www.epfl.ch/labs/mlo/igh-intelligent-global-health/'>intelligent Global Health</a>.). On a cohort of 400 Swiss patients suspected of LRTI, the model obtained over 90% area under the ROC curve for this task. <br><br>*Simplified to ensure smooth running on your browser, the performance is minimally affected. Details of the adaptations are below <br>- <b>Removed</b>: positional embedding (i.e. we don’t take the anatomic position into consideration). Rather, the model now does mean pooling over the feature vector of the images for each patient <br>- <b>Replaced</b>: ResNet18 by Mobilenet",
-                tradeoffs: 'We are using a simpler version of DeepChest in order to be able to run it on the browser.',
-                dataFormatInformation: 'This model takes as input an image dataset. It consists on a set of lung ultrasound images per patient with its corresponding label of covid positive or negative. Moreover, to identify the images per patient you have to follow the follwing naming pattern: "patientId_*.png"',
-                dataExampleText: 'Below you can find an example of an expected lung image for patient 2 named: 2_QAID_1.masked.reshaped.squared.224.png',
-                dataExampleImage: 'https://storage.googleapis.com/deai-313515.appspot.com/example_training_data/2_QAID_1.masked.reshaped.squared.224.png'
-            },
-            trainingInformation: {
-                modelID: 'lus-covid-model',
-                epochs: 15,
-                roundDuration: 10,
-                validationSplit: 0.2,
-                batchSize: 2,
-                modelCompileData: {
-                    optimizer: 'sgd',
-                    loss: 'binaryCrossentropy',
-                    metrics: ['accuracy']
-                },
-                learningRate: 0.001,
-                IMAGE_H: 100,
-                IMAGE_W: 100,
-                preprocessingFunctions: [__1.data.ImagePreprocessing.Resize],
-                LABEL_LIST: ['COVID-Positive', 'COVID-Negative'],
-                dataType: 'image',
-                scheme: 'Decentralized',
-                noiseScale: undefined,
-                clippingRadius: 20,
-                decentralizedSecure: true,
-                minimumReadyPeers: 3,
-                maxShareValue: 100
-            }
-        };
-    },
-    getModel: function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var imageHeight, imageWidth, imageChannels, numOutputClasses, model;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                imageHeight = 100;
-                imageWidth = 100;
-                imageChannels = 3;
-                numOutputClasses = 2;
-                model = __1.tf.sequential();
-                // In the first layer of our convolutional neural network we have
-                // to specify the input shape. Then we specify some parameters for
-                // the convolution operation that takes place in this layer.
-                model.add(__1.tf.layers.conv2d({
-                    inputShape: [imageHeight, imageWidth, imageChannels],
-                    kernelSize: 5,
-                    filters: 8,
-                    strides: 1,
-                    activation: 'relu',
-                    kernelInitializer: 'varianceScaling'
-                }));
-                // The MaxPooling layer acts as a sort of downsampling using max values
-                // in a region instead of averaging.
-                model.add(__1.tf.layers.maxPooling2d({ poolSize: [2, 2], strides: [2, 2] }));
-                // Repeat another conv2d + maxPooling stack.
-                // Note that we have more filters in the convolution.
-                model.add(__1.tf.layers.conv2d({
-                    kernelSize: 5,
-                    filters: 16,
-                    strides: 1,
-                    activation: 'relu',
-                    kernelInitializer: 'varianceScaling'
-                }));
-                model.add(__1.tf.layers.maxPooling2d({ poolSize: [2, 2], strides: [2, 2] }));
-                // Now we flatten the output from the 2D filters into a 1D vector to prepare
-                // it for input into our last layer. This is common practice when feeding
-                // higher dimensional data to a final classification output layer.
-                model.add(__1.tf.layers.flatten());
-                // Our last layer is a dense layer which has 2 output units, one for each
-                // output class.
-                model.add(__1.tf.layers.dense({
-                    units: numOutputClasses,
-                    kernelInitializer: 'varianceScaling',
-                    activation: 'softmax'
-                }));
-                return [2 /*return*/, model];
-            });
-        });
-    }
-};

package/dist/core/default_tasks/mnist.d.ts

@@ -1,2 +0,0 @@
-import { TaskProvider } from '..';
-export declare const mnist: TaskProvider;

package/dist/core/default_tasks/simple_face.d.ts

@@ -1,2 +0,0 @@
-import { TaskProvider } from '..';
-export declare const simpleFace: TaskProvider;

package/dist/core/default_tasks/titanic.d.ts

@@ -1,2 +0,0 @@
-import { TaskProvider } from '..';
-export declare const titanic: TaskProvider;

package/dist/core/index.d.ts

@@ -1,18 +0,0 @@
-export * as tf from '@tensorflow/tfjs';
-export * as data from './dataset';
-export * as serialization from './serialization';
-export * as training from './training';
-export * as privacy from './privacy';
-export { GraphInformant, TrainingInformant, informant } from './informant';
-export { Base as Client } from './client';
-export * as client from './client';
-export { WeightsContainer, aggregation } from './weights';
-export { AsyncBuffer } from './async_buffer';
-export { AsyncInformant } from './async_informant';
-export { Logger, ConsoleLogger, TrainerLog } from './logging';
-export { Memory, ModelType, ModelInfo, Path, ModelSource, Empty as EmptyMemory } from './memory';
-export { Disco, TrainingSchemes } from './training';
-export { Validator } from './validation';
-export * from './task';
-export * as defaultTasks from './default_tasks';
-export * from './types';

package/dist/core/index.js

@@ -1,39 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.defaultTasks = exports.Validator = exports.TrainingSchemes = exports.Disco = exports.EmptyMemory = exports.ModelType = exports.Memory = exports.TrainerLog = exports.ConsoleLogger = exports.Logger = exports.AsyncInformant = exports.AsyncBuffer = exports.aggregation = exports.WeightsContainer = exports.client = exports.Client = exports.informant = exports.TrainingInformant = exports.GraphInformant = exports.privacy = exports.training = exports.serialization = exports.data = exports.tf = void 0;
-var tslib_1 = require("tslib");
-exports.tf = (0, tslib_1.__importStar)(require("@tensorflow/tfjs"));
-exports.data = (0, tslib_1.__importStar)(require("./dataset"));
-exports.serialization = (0, tslib_1.__importStar)(require("./serialization"));
-exports.training = (0, tslib_1.__importStar)(require("./training"));
-exports.privacy = (0, tslib_1.__importStar)(require("./privacy"));
-var informant_1 = require("./informant");
-Object.defineProperty(exports, "GraphInformant", { enumerable: true, get: function () { return informant_1.GraphInformant; } });
-Object.defineProperty(exports, "TrainingInformant", { enumerable: true, get: function () { return informant_1.TrainingInformant; } });
-Object.defineProperty(exports, "informant", { enumerable: true, get: function () { return informant_1.informant; } });
-var client_1 = require("./client");
-Object.defineProperty(exports, "Client", { enumerable: true, get: function () { return client_1.Base; } });
-exports.client = (0, tslib_1.__importStar)(require("./client"));
-var weights_1 = require("./weights");
-Object.defineProperty(exports, "WeightsContainer", { enumerable: true, get: function () { return weights_1.WeightsContainer; } });
-Object.defineProperty(exports, "aggregation", { enumerable: true, get: function () { return weights_1.aggregation; } });
-var async_buffer_1 = require("./async_buffer");
-Object.defineProperty(exports, "AsyncBuffer", { enumerable: true, get: function () { return async_buffer_1.AsyncBuffer; } });
-var async_informant_1 = require("./async_informant");
-Object.defineProperty(exports, "AsyncInformant", { enumerable: true, get: function () { return async_informant_1.AsyncInformant; } });
-var logging_1 = require("./logging");
-Object.defineProperty(exports, "Logger", { enumerable: true, get: function () { return logging_1.Logger; } });
-Object.defineProperty(exports, "ConsoleLogger", { enumerable: true, get: function () { return logging_1.ConsoleLogger; } });
-Object.defineProperty(exports, "TrainerLog", { enumerable: true, get: function () { return logging_1.TrainerLog; } });
-var memory_1 = require("./memory");
-Object.defineProperty(exports, "Memory", { enumerable: true, get: function () { return memory_1.Memory; } });
-Object.defineProperty(exports, "ModelType", { enumerable: true, get: function () { return memory_1.ModelType; } });
-Object.defineProperty(exports, "EmptyMemory", { enumerable: true, get: function () { return memory_1.Empty; } });
-var training_1 = require("./training");
-Object.defineProperty(exports, "Disco", { enumerable: true, get: function () { return training_1.Disco; } });
-Object.defineProperty(exports, "TrainingSchemes", { enumerable: true, get: function () { return training_1.TrainingSchemes; } });
-var validation_1 = require("./validation");
-Object.defineProperty(exports, "Validator", { enumerable: true, get: function () { return validation_1.Validator; } });
-(0, tslib_1.__exportStar)(require("./task"), exports);
-exports.defaultTasks = (0, tslib_1.__importStar)(require("./default_tasks"));
-(0, tslib_1.__exportStar)(require("./types"), exports);

package/dist/core/informant/graph_informant.js

@@ -1,23 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GraphInformant = void 0;
-var immutable_1 = require("immutable");
-var GraphInformant = /** @class */ (function () {
-    function GraphInformant() {
-        this.currentAccuracy = 0;
-        this.accuracyDataSeries = (0, immutable_1.Repeat)(0, GraphInformant.NB_EPOCHS_ON_GRAPH).toList();
-    }
-    GraphInformant.prototype.updateAccuracy = function (accuracy) {
-        this.accuracyDataSeries = this.accuracyDataSeries.shift().push(accuracy);
-        this.currentAccuracy = accuracy;
-    };
-    GraphInformant.prototype.data = function () {
-        return this.accuracyDataSeries;
-    };
-    GraphInformant.prototype.accuracy = function () {
-        return this.currentAccuracy;
-    };
-    GraphInformant.NB_EPOCHS_ON_GRAPH = 10;
-    return GraphInformant;
-}());
-exports.GraphInformant = GraphInformant;

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

@@ -1,3 +0,0 @@
-export { GraphInformant } from './graph_informant';
-export { Base as TrainingInformant } from './training_informant';
-export * as informant from './training_informant';

package/dist/core/informant/index.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.informant = exports.TrainingInformant = exports.GraphInformant = void 0;
-var tslib_1 = require("tslib");
-var graph_informant_1 = require("./graph_informant");
-Object.defineProperty(exports, "GraphInformant", { enumerable: true, get: function () { return graph_informant_1.GraphInformant; } });
-var training_informant_1 = require("./training_informant");
-Object.defineProperty(exports, "TrainingInformant", { enumerable: true, get: function () { return training_informant_1.Base; } });
-exports.informant = (0, tslib_1.__importStar)(require("./training_informant"));

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

@@ -1,31 +0,0 @@
-import { List } from 'immutable';
-import { Task } from '../../task';
-import { GraphInformant } from '../graph_informant';
-export declare abstract class Base {
-    readonly task: Task;
-    private readonly nbrMessagesToShow;
-    private messages;
-    protected readonly trainingGraphInformant: GraphInformant;
-    protected readonly validationGraphInformant: GraphInformant;
-    protected currentRound: number;
-    protected currentNumberOfParticipants: number;
-    protected totalNumberOfParticipants: number;
-    protected averageNumberOfParticipants: number;
-    constructor(task: Task, nbrMessagesToShow?: number);
-    abstract update(statistics: Record<string, number>): void;
-    addMessage(msg: string): void;
-    getMessages(): string[];
-    round(): number;
-    participants(): number;
-    totalParticipants(): number;
-    averageParticipants(): number;
-    updateTrainingGraph(accuracy: number): void;
-    updateValidationGraph(accuracy: number): void;
-    trainingAccuracy(): number;
-    validationAccuracy(): number;
-    trainingAccuracyData(): List<number>;
-    validationAccuracyData(): List<number>;
-    isDecentralized(): boolean;
-    isFederated(): boolean;
-    static isTrainingInformant(raw: unknown): raw is Base;
-}

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.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.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;