@epfml/discojs 1.0.0 → 2.0.0

package/dist/tfjs.js

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

package/dist/training/disco.d.ts

@@ -1,14 +0,0 @@
-import { Memory } from '@/memory';
-import { Client, dataset, Logger, Task, TrainingInformant, TrainingSchemes } from '..';
-import { TrainerLog } from '../logging/trainer_logger';
-export declare class Disco {
-    readonly task: Task;
-    readonly logger: Logger;
-    readonly memory: Memory;
-    private readonly client;
-    private readonly trainer;
-    constructor(task: Task, logger: Logger, memory: Memory, scheme: TrainingSchemes, informant: TrainingInformant, client: Client);
-    startTraining(dataTuple: dataset.DataTuple): Promise<void>;
-    stopTraining(): Promise<void>;
-    getTrainerLog(): Promise<TrainerLog>;
-}

package/dist/training/disco.js

@@ -1,70 +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, logger, memory, scheme, informant, client) {
-        this.task = task;
-        this.logger = logger;
-        this.memory = memory;
-        this.client = client;
-        if (client.task !== task) {
-            throw new Error('client not setup for given task');
-        }
-        if (informant.taskID !== task.taskID) {
-            throw new Error('informant not setup for given task');
-        }
-        var trainerBuilder = new trainer_builder_1.TrainerBuilder(this.memory, this.task, informant);
-        this.trainer = trainerBuilder.build(this.client, scheme !== __1.TrainingSchemes.LOCAL);
-    }
-    Disco.prototype.startTraining = function (dataTuple) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var 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');
-                        valDataset = dataTuple.validation !== undefined ? dataTuple.validation.dataset : dataTuple.train.dataset;
-                        return [4 /*yield*/, this.trainer];
-                    case 1: return [4 /*yield*/, (_a.sent()).trainModel(dataTuple.train.dataset, valDataset)];
-                    case 2:
-                        _a.sent();
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    // Stops the training function
-    //
-    // do not disconnect given client
-    Disco.prototype.stopTraining = 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.getTrainerLog = 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/training/training_schemes.d.ts

@@ -1,5 +0,0 @@
-export declare enum TrainingSchemes {
-    LOCAL = "local",
-    DECENTRALIZED = "deai",
-    FEDERATED = "feai"
-}