@epfml/discojs-node 2.0.0 → 2.1.2-p20240506085037.0

package/dist/core/tasks/index.js

@@ -1,10 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.geotags = exports.simple_face = exports.titanic = exports.mnist = exports.lus_covid = exports.cifar10 = void 0;
-var tslib_1 = require("tslib");
-exports.cifar10 = (0, tslib_1.__importStar)(require("./cifar10"));
-exports.lus_covid = (0, tslib_1.__importStar)(require("./lus_covid"));
-exports.mnist = (0, tslib_1.__importStar)(require("./mnist"));
-exports.titanic = (0, tslib_1.__importStar)(require("./titanic"));
-exports.simple_face = (0, tslib_1.__importStar)(require("./simple_face"));
-exports.geotags = (0, tslib_1.__importStar)(require("./geotags"));

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

@@ -1,3 +0,0 @@
-import { tf, Task } from '..';
-export declare const task: Task;
-export declare function model(): tf.LayersModel;

package/dist/core/tasks/lus_covid.js

@@ -1,87 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.model = exports.task = void 0;
-var __1 = require("..");
-exports.task = {
-    taskID: 'lus_covid',
-    displayInformation: {
-        taskTitle: 'COVID Lung Ultrasound',
-        summary: {
-            preview: 'Do you have a dataset 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
-    }
-};
-function model() {
-    var imageHeight = 100;
-    var imageWidth = 100;
-    var imageChannels = 3;
-    var numOutputClasses = 2;
-    var 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 model;
-}
-exports.model = model;

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

@@ -1,3 +0,0 @@
-import { tf, Task } from '..';
-export declare const task: Task;
-export declare function model(): tf.LayersModel;

package/dist/core/tasks/mnist.js

@@ -1,60 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.model = exports.task = void 0;
-var __1 = require("..");
-exports.task = {
-    taskID: 'mnist',
-    displayInformation: {
-        taskTitle: 'MNIST',
-        summary: {
-            preview: "Test our platform by using a publicly available <b>image</b> dataset. <br><br> Download the classic MNIST imagebank of hand-written numbers <a class='underline text-primary-dark dark:text-primary-light' href='https://www.kaggle.com/scolianni/mnistasjpg'>here</a>. <br> This model learns to identify hand written numbers.",
-            overview: 'The MNIST handwritten digit classification problem is a standard dataset used in computer vision and deep learning. Although the dataset is effectively solved, we use it to test our Decentralised Learning algorithms and platform.'
-        },
-        model: 'The current model is a very simple CNN and its main goal is to test the app and the Decentralizsed Learning functionality.',
-        tradeoffs: 'We are using a simple model, first a 2d convolutional layer > max pooling > 2d convolutional layer > max pooling > convolutional layer > 2 dense layers.',
-        dataFormatInformation: 'This model is trained on images corresponding to digits 0 to 9. You can upload each digit image of your dataset in the box corresponding to its label. The model taskes images of size 28x28 as input.',
-        dataExampleText: 'Below you can find an example of an expected image representing the digit 9.',
-        dataExampleImage: 'http://storage.googleapis.com/deai-313515.appspot.com/example_training_data/9-mnist-example.png'
-    },
-    trainingInformation: {
-        modelID: 'mnist-model',
-        epochs: 10,
-        roundDuration: 10,
-        validationSplit: 0.2,
-        batchSize: 30,
-        modelCompileData: {
-            optimizer: 'rmsprop',
-            loss: 'categoricalCrossentropy',
-            metrics: ['accuracy']
-        },
-        dataType: 'image',
-        IMAGE_H: 28,
-        IMAGE_W: 28,
-        preprocessingFunctions: [],
-        LABEL_LIST: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'],
-        scheme: 'Decentralized',
-        noiseScale: undefined,
-        clippingRadius: 20,
-        decentralizedSecure: true,
-        minimumReadyPeers: 3,
-        maxShareValue: 100
-    }
-};
-function model() {
-    var model = __1.tf.sequential();
-    model.add(__1.tf.layers.conv2d({
-        inputShape: [28, 28, 3],
-        kernelSize: 3,
-        filters: 16,
-        activation: 'relu'
-    }));
-    model.add(__1.tf.layers.maxPooling2d({ poolSize: 2, strides: 2 }));
-    model.add(__1.tf.layers.conv2d({ kernelSize: 3, filters: 32, activation: 'relu' }));
-    model.add(__1.tf.layers.maxPooling2d({ poolSize: 2, strides: 2 }));
-    model.add(__1.tf.layers.conv2d({ kernelSize: 3, filters: 32, activation: 'relu' }));
-    model.add(__1.tf.layers.flatten({}));
-    model.add(__1.tf.layers.dense({ units: 64, activation: 'relu' }));
-    model.add(__1.tf.layers.dense({ units: 10, activation: 'softmax' }));
-    return model;
-}
-exports.model = model;

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

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

package/dist/core/tasks/simple_face.js

@@ -1,41 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.task = void 0;
-var __1 = require("..");
-exports.task = {
-    taskID: 'simple_face',
-    displayInformation: {
-        taskTitle: 'Simple Face',
-        summary: {
-            preview: 'Can you detect if the person in a picture is a child or an adult?',
-            overview: 'Simple face is a small subset of face_task from Kaggle'
-        },
-        limitations: 'The training data is limited to small images of size 200x200.',
-        tradeoffs: 'Training success strongly depends on label distribution',
-        dataFormatInformation: '',
-        dataExampleText: 'Below you find an example',
-        dataExampleImage: 'https://storage.googleapis.com/deai-313515.appspot.com/example_training_data/simple_face-example.png'
-    },
-    trainingInformation: {
-        modelID: 'simple_face-model',
-        epochs: 50,
-        modelURL: 'https://storage.googleapis.com/deai-313515.appspot.com/models/mobileNetV2_35_alpha_2_classes/model.json',
-        roundDuration: 1,
-        validationSplit: 0.2,
-        batchSize: 10,
-        preprocessingFunctions: [__1.data.ImagePreprocessing.Normalize],
-        learningRate: 0.001,
-        modelCompileData: {
-            optimizer: 'sgd',
-            loss: 'categoricalCrossentropy',
-            metrics: ['accuracy']
-        },
-        dataType: 'image',
-        IMAGE_H: 200,
-        IMAGE_W: 200,
-        LABEL_LIST: ['child', 'adult'],
-        scheme: 'Federated',
-        noiseScale: undefined,
-        clippingRadius: undefined
-    }
-};

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

@@ -1,3 +0,0 @@
-import { tf, Task } from '..';
-export declare const task: Task;
-export declare function model(): tf.LayersModel;

package/dist/core/tasks/titanic.js

@@ -1,88 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.model = exports.task = void 0;
-var __1 = require("..");
-exports.task = {
-    taskID: 'titanic',
-    displayInformation: {
-        taskTitle: 'Titanic',
-        summary: {
-            preview: "Test our platform by using a publicly available <b>tabular</b> dataset. <br><br> Download the passenger list from the Titanic shipwreck here: <a class='underline text-primary-dark dark:text-primary-light' href='https://github.com/epfml/disco/raw/develop/example_training_data/titanic_train.csv'>titanic_train.csv</a> (more info <a class='underline text-primary-dark dark:text-primary-light' href='https://www.kaggle.com/c/titanic'>here</a>). <br> This model predicts the type of person most likely to survive/die in the historic ship accident, based on their characteristics (sex, age, class etc.).",
-            overview: 'We all know the unfortunate story of the Titanic: this flamboyant new transatlantic boat that sunk in 1912 in the North Atlantic Ocean. Today, we revist this tragedy by trying to predict the survival odds of the passenger given some basic features.'
-        },
-        model: 'The current model does not normalize the given data and applies only a very simple pre-processing of the data.',
-        tradeoffs: 'We are using a small model for this task: 4 fully connected layers with few neurons. This allows fast training but can yield to reduced accuracy.',
-        dataFormatInformation: 'This model takes as input a CSV file with 12 columns. The features are general information about the passenger (sex, age, name, etc.) and specific related Titanic data such as the ticket class bought by the passenger, its cabin number, etc.<br><br>pclass: A proxy for socio-economic status (SES)<br>1st = Upper<br>2nd = Middle<br>3rd = Lower<br><br>age: Age is fractional if less than 1. If the age is estimated, it is in the form of xx.5<br><br>sibsp: The dataset defines family relations in this way:<br>Sibling = brother, sister, stepbrother, stepsister<br>Spouse = husband, wife (mistresses and fiancés were ignored)<br><br>parch: The dataset defines family relations in this way:<br>Parent = mother, father<br>Child = daughter, son, stepdaughter, stepson<br>Some children travelled only with a nanny, therefore parch=0 for them.<br><br>The first line of the CSV contains the header:<br> PassengerId, Survived, Pclass, Name, Sex, Age, SibSp, Parch, Ticket, Fare, Cabin, Embarked<br><br>Each susequent row contains the corresponding data.',
-        dataExampleText: 'Below one can find an example of a datapoint taken as input by our model. In this datapoint, the person is young man named Owen Harris that unfortunnalty perished with the Titanic. He boarded the boat in South Hamptons and was a 3rd class passenger. On the testing & validation page, the data should not contain the label column (Survived).',
-        dataExample: [
-            { columnName: 'PassengerId', columnData: '1' },
-            { columnName: 'Survived', columnData: '0' },
-            { columnName: 'Name', columnData: 'Braund, Mr. Owen Harris' },
-            { columnName: 'Sex', columnData: 'male' },
-            { columnName: 'Age', columnData: '22' },
-            { columnName: 'SibSp', columnData: '1' },
-            { columnName: 'Parch', columnData: '0' },
-            { columnName: 'Ticket', columnData: '1/5 21171' },
-            { columnName: 'Fare', columnData: '7.25' },
-            { columnName: 'Cabin', columnData: 'E46' },
-            { columnName: 'Embarked', columnData: 'S' },
-            { columnName: 'Pclass', columnData: '3' }
-        ],
-        headers: [
-            'PassengerId',
-            'Survived',
-            'Name',
-            'Sex',
-            'Age',
-            'SibSp',
-            'Parch',
-            'Ticket',
-            'Fare',
-            'Cabin',
-            'Embarked',
-            'Pclass'
-        ]
-    },
-    trainingInformation: {
-        modelID: 'titanic-model',
-        epochs: 20,
-        roundDuration: 10,
-        validationSplit: 0,
-        batchSize: 30,
-        preprocessingFunctions: [],
-        modelCompileData: {
-            optimizer: 'rmsprop',
-            loss: 'binaryCrossentropy',
-            metrics: ['accuracy']
-        },
-        dataType: 'tabular',
-        inputColumns: [
-            'PassengerId',
-            'Age',
-            'SibSp',
-            'Parch',
-            'Fare',
-            'Pclass'
-        ],
-        outputColumns: [
-            'Survived'
-        ],
-        scheme: 'Federated',
-        noiseScale: undefined,
-        clippingRadius: undefined
-    }
-};
-function model() {
-    var model = __1.tf.sequential();
-    model.add(__1.tf.layers.dense({
-        inputShape: [6],
-        units: 124,
-        activation: 'relu',
-        kernelInitializer: 'leCunNormal'
-    }));
-    model.add(__1.tf.layers.dense({ units: 64, activation: 'relu' }));
-    model.add(__1.tf.layers.dense({ units: 32, activation: 'relu' }));
-    model.add(__1.tf.layers.dense({ units: 1, activation: 'sigmoid' }));
-    return model;
-}
-exports.model = model;

package/dist/core/training/disco.d.ts

@@ -1,23 +0,0 @@
-import { Client, data, Logger, Task, TrainingInformant, TrainingSchemes, Memory } from '..';
-import { TrainerLog } from '../logging/trainer_logger';
-interface DiscoOptions {
-    client?: Client;
-    url?: string | URL;
-    scheme?: TrainingSchemes;
-    informant?: TrainingInformant;
-    logger?: Logger;
-    memory?: Memory;
-}
-export declare class Disco {
-    readonly task: Task;
-    readonly logger: Logger;
-    readonly memory: Memory;
-    private readonly client;
-    private readonly trainer;
-    constructor(task: Task, options: DiscoOptions);
-    fit(dataTuple: data.DataSplit): Promise<void>;
-    pause(): Promise<void>;
-    close(): Promise<void>;
-    logs(): Promise<TrainerLog>;
-}
-export {};

package/dist/core/training/disco.js

@@ -1,130 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Disco = void 0;
-var tslib_1 = require("tslib");
-var __1 = require("..");
-var trainer_builder_1 = require("./trainer/trainer_builder");
-// Handles the training loop, server communication & provides the user with feedback.
-var Disco = /** @class */ (function () {
-    // client need to be connected
-    function Disco(task, options) {
-        if (options.scheme === undefined) {
-            options.scheme = __1.TrainingSchemes[task.trainingInformation.scheme];
-        }
-        if (options.client === undefined) {
-            if (options.url === undefined) {
-                throw new Error('could not determine client from given parameters');
-            }
-            if (typeof options.url === 'string') {
-                options.url = new URL(options.url);
-            }
-            switch (options.scheme) {
-                case __1.TrainingSchemes.FEDERATED:
-                    options.client = new __1.client.federated.Client(options.url, task);
-                    break;
-                case __1.TrainingSchemes.DECENTRALIZED:
-                    options.client = new __1.client.federated.Client(options.url, task);
-                    break;
-                default:
-                    options.client = new __1.client.Local(options.url, task);
-                    break;
-            }
-        }
-        if (options.informant === undefined) {
-            switch (options.scheme) {
-                case __1.TrainingSchemes.FEDERATED:
-                    options.informant = new __1.informant.FederatedInformant(task);
-                    break;
-                case __1.TrainingSchemes.DECENTRALIZED:
-                    options.informant = new __1.informant.DecentralizedInformant(task);
-                    break;
-                default:
-                    options.informant = new __1.informant.LocalInformant(task);
-                    break;
-            }
-        }
-        if (options.logger === undefined) {
-            options.logger = new __1.ConsoleLogger();
-        }
-        if (options.memory === undefined) {
-            options.memory = new __1.EmptyMemory();
-        }
-        if (options.client.task !== task) {
-            throw new Error('client not setup for given task');
-        }
-        if (options.informant.task.taskID !== task.taskID) {
-            throw new Error('informant not setup for given task');
-        }
-        this.task = task;
-        this.client = options.client;
-        this.memory = options.memory;
-        this.logger = options.logger;
-        var trainerBuilder = new trainer_builder_1.TrainerBuilder(this.memory, this.task, options.informant);
-        this.trainer = trainerBuilder.build(this.client, options.scheme !== __1.TrainingSchemes.LOCAL);
-    }
-    Disco.prototype.fit = function (dataTuple) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var trainDataset, valDataset;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        this.logger.success('Thank you for your contribution. Data preprocessing has started');
-                        trainDataset = dataTuple.train.batch().preprocess();
-                        valDataset = dataTuple.validation !== undefined
-                            ? dataTuple.validation.batch().preprocess()
-                            : trainDataset;
-                        return [4 /*yield*/, this.client.connect()];
-                    case 1:
-                        _a.sent();
-                        return [4 /*yield*/, this.trainer];
-                    case 2: return [4 /*yield*/, (_a.sent()).trainModel(trainDataset.dataset, valDataset.dataset)];
-                    case 3:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    // Stops the training function. Does not disconnect the client.
-    Disco.prototype.pause = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.trainer];
-                    case 1: return [4 /*yield*/, (_a.sent()).stopTraining()];
-                    case 2:
-                        _a.sent();
-                        this.logger.success('Training was successfully interrupted.');
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    Disco.prototype.close = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.pause()];
-                    case 1:
-                        _a.sent();
-                        return [4 /*yield*/, this.client.disconnect()];
-                    case 2:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    Disco.prototype.logs = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.trainer];
-                    case 1: return [2 /*return*/, (_a.sent()).getTrainerLog()];
-                }
-            });
-        });
-    };
-    return Disco;
-}());
-exports.Disco = Disco;

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

@@ -1,2 +0,0 @@
-export { Disco } from './disco';
-export { TrainingSchemes } from './training_schemes';

package/dist/core/training/index.js

@@ -1,7 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TrainingSchemes = exports.Disco = void 0;
-var disco_1 = require("./disco");
-Object.defineProperty(exports, "Disco", { enumerable: true, get: function () { return disco_1.Disco; } });
-var training_schemes_1 = require("./training_schemes");
-Object.defineProperty(exports, "TrainingSchemes", { enumerable: true, get: function () { return training_schemes_1.TrainingSchemes; } });

package/dist/core/training/trainer/distributed_trainer.d.ts

@@ -1,20 +0,0 @@
-import { tf, Client, Memory, Task, TrainingInformant } from '../..';
-import { Trainer } from './trainer';
-/**
- * Class whose role is to train a model in a distributed way with a given dataset.
- */
-export declare class DistributedTrainer extends Trainer {
-    private readonly previousRoundModel;
-    private readonly client;
-    /** DistributedTrainer constructor, accepts same arguments as Trainer and in additional also a client who takes care of communicating weights.
-     */
-    constructor(task: Task, trainingInformant: TrainingInformant, memory: Memory, model: tf.LayersModel, previousRoundModel: tf.LayersModel, client: Client);
-    /**
-     * Callback called every time a round is over
-     */
-    onRoundEnd(accuracy: number): Promise<void>;
-    /**
-     * Callback called once training is over
-     */
-    onTrainEnd(): Promise<void>;
-}

package/dist/core/training/trainer/distributed_trainer.js

@@ -1,65 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DistributedTrainer = void 0;
-var tslib_1 = require("tslib");
-var __1 = require("../..");
-var trainer_1 = require("./trainer");
-/**
- * Class whose role is to train a model in a distributed way with a given dataset.
- */
-var DistributedTrainer = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(DistributedTrainer, _super);
-    /** DistributedTrainer constructor, accepts same arguments as Trainer and in additional also a client who takes care of communicating weights.
-     */
-    function DistributedTrainer(task, trainingInformant, memory, model, previousRoundModel, client) {
-        var _this = _super.call(this, task, trainingInformant, memory, model) || this;
-        _this.previousRoundModel = previousRoundModel;
-        _this.client = client;
-        return _this;
-    }
-    /**
-     * Callback called every time a round is over
-     */
-    DistributedTrainer.prototype.onRoundEnd = function (accuracy) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var currentRoundWeights, previousRoundWeights, aggregatedWeights;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        currentRoundWeights = __1.WeightsContainer.from(this.model);
-                        previousRoundWeights = __1.WeightsContainer.from(this.previousRoundModel);
-                        return [4 /*yield*/, this.client.onRoundEndCommunication(currentRoundWeights, previousRoundWeights, this.roundTracker.round, this.trainingInformant)];
-                    case 1:
-                        aggregatedWeights = _a.sent();
-                        this.previousRoundModel.setWeights(currentRoundWeights.weights);
-                        this.model.setWeights(aggregatedWeights.weights);
-                        return [4 /*yield*/, this.memory.updateWorkingModel({ taskID: this.task.taskID, name: this.trainingInformation.modelID }, this.model)];
-                    case 2:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    // if it is undefined, will training continue? we hope yes
-    /**
-     * Callback called once training is over
-     */
-    DistributedTrainer.prototype.onTrainEnd = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.client.onTrainEndCommunication(__1.WeightsContainer.from(this.model), this.trainingInformant)];
-                    case 1:
-                        _a.sent();
-                        return [4 /*yield*/, _super.prototype.onTrainEnd.call(this)];
-                    case 2:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    return DistributedTrainer;
-}(trainer_1.Trainer));
-exports.DistributedTrainer = DistributedTrainer;

package/dist/core/training/trainer/local_trainer.d.ts

@@ -1,11 +0,0 @@
-import { tf } from '../..';
-import { Trainer } from './trainer';
-/** Class whose role is to locally (alone) train a model on a given dataset, without any collaborators.
- */
-export declare class LocalTrainer extends Trainer {
-    /**
-     * Callback called every time a round is over. For local training, a round is typically an epoch
-     */
-    onRoundEnd(accuracy: number): Promise<void>;
-    protected onEpochEnd(epoch: number, logs?: tf.Logs): void;
-}

package/dist/core/training/trainer/local_trainer.js

@@ -1,34 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.LocalTrainer = void 0;
-var tslib_1 = require("tslib");
-var trainer_1 = require("./trainer");
-/** Class whose role is to locally (alone) train a model on a given dataset, without any collaborators.
- */
-var LocalTrainer = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(LocalTrainer, _super);
-    function LocalTrainer() {
-        return _super !== null && _super.apply(this, arguments) || this;
-    }
-    /**
-     * Callback called every time a round is over. For local training, a round is typically an epoch
-     */
-    LocalTrainer.prototype.onRoundEnd = function (accuracy) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, this.memory.updateWorkingModel({ taskID: this.task.taskID, name: this.trainingInformation.modelID }, this.model)];
-                    case 1:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    LocalTrainer.prototype.onEpochEnd = function (epoch, logs) {
-        _super.prototype.onEpochEnd.call(this, epoch, logs);
-        this.trainingInformant.update({ currentRound: epoch });
-    };
-    return LocalTrainer;
-}(trainer_1.Trainer));
-exports.LocalTrainer = LocalTrainer;

package/dist/core/training/trainer/round_tracker.d.ts

@@ -1,30 +0,0 @@
-/**
- * Class that keeps track of the current batch in order for the trainer to query when round has ended.
- *
- * @remark
- * In distributed training, the client trains locally for a certain amount of epochs before sharing his weights to the server/neighbor, this
- * is what we call a round.
- *
- * The role of the RoundTracker is to keep track of when a roundHasEnded using the current batch number. The batch in the RoundTracker is cumulative whereas
- * in the onBatchEnd it is not (it resets to 0 after each epoch).
- *
- * The roundDuration is the length of a round (in batches).
- */
-export declare class RoundTracker {
-    round: number;
-    batch: number;
-    roundDuration: number;
-    constructor(roundDuration: number);
-    /**
-     * Update the batch number, to be called inside onBatchEnd. (We do not use batch output of onBatchEnd since it is
-     * not cumulative).
-     */
-    updateBatch(): void;
-    /**
-     * Returns true if a local round has ended, false otherwise.
-     *
-     * @remark
-     * Returns true if (batch) mod (batches per round) == 0, false otherwise
-     */
-    roundHasEnded(): boolean;
-}