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

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

package/dist/core/serialization/model.js

@@ -1,55 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.decode = exports.encode = exports.isEncoded = void 0;
-var tslib_1 = require("tslib");
-var __1 = require("..");
-var msgpack_lite_1 = (0, tslib_1.__importDefault)(require("msgpack-lite"));
-function isEncoded(raw) {
-    return Array.isArray(raw) && raw.every(function (r) { return typeof r === 'number'; });
-}
-exports.isEncoded = isEncoded;
-function encode(model) {
-    return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-        var saved;
-        var _this = this;
-        return (0, tslib_1.__generator)(this, function (_a) {
-            switch (_a.label) {
-                case 0: return [4 /*yield*/, new Promise(function (resolve) {
-                        void model.save({
-                            save: function (artifacts) { return (0, tslib_1.__awaiter)(_this, void 0, void 0, function () {
-                                return (0, tslib_1.__generator)(this, function (_a) {
-                                    resolve(artifacts);
-                                    return [2 /*return*/, {
-                                            modelArtifactsInfo: {
-                                                dateSaved: new Date(),
-                                                modelTopologyType: 'JSON'
-                                            }
-                                        }];
-                                });
-                            }); }
-                        });
-                    })];
-                case 1:
-                    saved = _a.sent();
-                    return [2 /*return*/, (0, tslib_1.__spreadArray)([], (0, tslib_1.__read)(msgpack_lite_1.default.encode(saved).values()), false)];
-            }
-        });
-    });
-}
-exports.encode = encode;
-function decode(encoded) {
-    return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-        var raw;
-        return (0, tslib_1.__generator)(this, function (_a) {
-            switch (_a.label) {
-                case 0:
-                    raw = msgpack_lite_1.default.decode(encoded);
-                    return [4 /*yield*/, __1.tf.loadLayersModel({
-                            load: function () { return raw; }
-                        })];
-                case 1: return [2 /*return*/, _a.sent()];
-            }
-        });
-    });
-}
-exports.decode = decode;

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

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

package/dist/core/serialization/weights.js

@@ -1,64 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.decode = exports.encode = exports.isEncoded = void 0;
-var tslib_1 = require("tslib");
-var msgpack = (0, tslib_1.__importStar)(require("msgpack-lite"));
-var __1 = require("..");
-function isSerialized(raw) {
-    if (typeof raw !== 'object' || raw === null) {
-        return false;
-    }
-    if (!('shape' in raw && 'data' in raw)) {
-        return false;
-    }
-    var _a = raw, shape = _a.shape, data = _a.data;
-    if (!(Array.isArray(shape) && shape.every(function (e) { return typeof e === 'number'; })) ||
-        !(Array.isArray(data) && data.every(function (e) { return typeof e === 'number'; }))) {
-        return false;
-    }
-    // eslint-disable-next-line
-    var _ = { shape: shape, data: data };
-    return true;
-}
-function isEncoded(raw) {
-    return Array.isArray(raw) && raw.every(function (e) { return typeof e === 'number'; });
-}
-exports.isEncoded = isEncoded;
-function encode(weights) {
-    return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-        var serialized;
-        var _this = this;
-        return (0, tslib_1.__generator)(this, function (_a) {
-            switch (_a.label) {
-                case 0: return [4 /*yield*/, Promise.all(weights.weights.map(function (t) { return (0, tslib_1.__awaiter)(_this, void 0, void 0, function () {
-                        var _a;
-                        var _b;
-                        return (0, tslib_1.__generator)(this, function (_c) {
-                            switch (_c.label) {
-                                case 0:
-                                    _b = {
-                                        shape: t.shape
-                                    };
-                                    _a = [[]];
-                                    return [4 /*yield*/, t.data()];
-                                case 1: return [2 /*return*/, (_b.data = tslib_1.__spreadArray.apply(void 0, _a.concat([tslib_1.__read.apply(void 0, [_c.sent()]), false])),
-                                        _b)];
-                            }
-                        });
-                    }); }))];
-                case 1:
-                    serialized = _a.sent();
-                    return [2 /*return*/, (0, tslib_1.__spreadArray)([], (0, tslib_1.__read)(msgpack.encode(serialized).values()), false)];
-            }
-        });
-    });
-}
-exports.encode = encode;
-function decode(encoded) {
-    var raw = msgpack.decode(encoded);
-    if (!(Array.isArray(raw) && raw.every(isSerialized))) {
-        throw new Error('expected to decode an array of serialized weights');
-    }
-    return new __1.WeightsContainer(raw.map(function (w) { return __1.tf.tensor(w.data, w.shape); }));
-}
-exports.decode = decode;

package/dist/core/task/data_example.d.ts

@@ -1,5 +0,0 @@
-export declare function isDataExample(raw: unknown): raw is DataExample;
-export interface DataExample {
-    columnName: string;
-    columnData: string | number;
-}

package/dist/core/task/data_example.js

@@ -1,24 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isDataExample = void 0;
-var immutable_1 = require("immutable");
-function isDataExample(raw) {
-    if (typeof raw !== 'object') {
-        return false;
-    }
-    if (raw === null) {
-        return false;
-    }
-    if (!(0, immutable_1.Set)(Object.keys(raw)).equals(immutable_1.Set.of('columnName', 'columnData'))) {
-        return false;
-    }
-    var _a = raw, columnName = _a.columnName, columnData = _a.columnData;
-    if (typeof columnName !== 'string' ||
-        (typeof columnData !== 'string' && typeof columnData !== 'number')) {
-        return false;
-    }
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    var _ = { columnName: columnName, columnData: columnData };
-    return true;
-}
-exports.isDataExample = isDataExample;

package/dist/core/task/display_information.d.ts

@@ -1,15 +0,0 @@
-import { Summary } from './summary';
-import { DataExample } from './data_example';
-export declare function isDisplayInformation(raw: unknown): raw is DisplayInformation;
-export interface DisplayInformation {
-    taskTitle?: string;
-    summary?: Summary;
-    tradeoffs?: string;
-    dataFormatInformation?: string;
-    dataExampleText?: string;
-    model?: string;
-    dataExample?: DataExample[];
-    headers?: string[];
-    dataExampleImage?: string;
-    limitations?: string;
-}

package/dist/core/task/display_information.js

@@ -1,49 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isDisplayInformation = void 0;
-var summary_1 = require("./summary");
-var data_example_1 = require("./data_example");
-function isDisplayInformation(raw) {
-    if (typeof raw !== 'object') {
-        return false;
-    }
-    if (raw === null) {
-        return false;
-    }
-    var _a = raw, dataExample = _a.dataExample, dataExampleImage = _a.dataExampleImage, dataExampleText = _a.dataExampleText, dataFormatInformation = _a.dataFormatInformation, headers = _a.headers, limitations = _a.limitations, model = _a.model, summary = _a.summary, taskTitle = _a.taskTitle, tradeoffs = _a.tradeoffs;
-    if (typeof taskTitle !== 'string' ||
-        (dataExampleText !== undefined && typeof dataExampleText !== 'string') ||
-        (dataFormatInformation !== undefined && typeof dataFormatInformation !== 'string') ||
-        (tradeoffs !== undefined && typeof tradeoffs !== 'string') ||
-        (model !== undefined && typeof model !== 'string') ||
-        (dataExampleImage !== undefined && typeof dataExampleImage !== 'string') ||
-        (limitations !== undefined && typeof limitations !== 'string')) {
-        return false;
-    }
-    if (summary !== undefined && !(0, summary_1.isSummary)(summary)) {
-        return false;
-    }
-    if (dataExample !== undefined && !(Array.isArray(dataExample) &&
-        dataExample.every(data_example_1.isDataExample))) {
-        return false;
-    }
-    if (headers !== undefined && !(Array.isArray(headers) &&
-        headers.every(function (e) { return typeof e === 'string'; }))) {
-        return false;
-    }
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    var _ = {
-        taskTitle: taskTitle,
-        summary: summary,
-        tradeoffs: tradeoffs,
-        dataFormatInformation: dataFormatInformation,
-        dataExampleText: dataExampleText,
-        model: model,
-        dataExample: dataExample,
-        headers: headers,
-        dataExampleImage: dataExampleImage,
-        limitations: limitations
-    };
-    return true;
-}
-exports.isDisplayInformation = isDisplayInformation;

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

@@ -1,3 +0,0 @@
-export { isTask, Task, isTaskID, TaskID } from './task';
-export { isDisplayInformation, DisplayInformation } from './display_information';
-export { TrainingInformation } from './training_information';

package/dist/core/task/index.js

@@ -1,8 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isDisplayInformation = exports.isTaskID = exports.isTask = void 0;
-var task_1 = require("./task");
-Object.defineProperty(exports, "isTask", { enumerable: true, get: function () { return task_1.isTask; } });
-Object.defineProperty(exports, "isTaskID", { enumerable: true, get: function () { return task_1.isTaskID; } });
-var display_information_1 = require("./display_information");
-Object.defineProperty(exports, "isDisplayInformation", { enumerable: true, get: function () { return display_information_1.isDisplayInformation; } });

package/dist/core/task/model_compile_data.d.ts

@@ -1,6 +0,0 @@
-export declare function isModelCompileData(raw: unknown): raw is ModelCompileData;
-export interface ModelCompileData {
-    optimizer: string;
-    loss: string;
-    metrics: string[];
-}

package/dist/core/task/model_compile_data.js

@@ -1,22 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isModelCompileData = void 0;
-function isModelCompileData(raw) {
-    if (typeof raw !== 'object') {
-        return false;
-    }
-    if (raw === null) {
-        return false;
-    }
-    var _a = raw, optimizer = _a.optimizer, loss = _a.loss, metrics = _a.metrics;
-    if (typeof optimizer !== 'string' ||
-        typeof loss !== 'string') {
-        return false;
-    }
-    if (!(Array.isArray(metrics) &&
-        metrics.every(function (e) { return typeof e === 'string'; }))) {
-        return false;
-    }
-    return true;
-}
-exports.isModelCompileData = isModelCompileData;

package/dist/core/task/summary.d.ts

@@ -1,5 +0,0 @@
-export declare function isSummary(raw: unknown): raw is Summary;
-export interface Summary {
-    preview: string;
-    overview: string;
-}

package/dist/core/task/summary.js

@@ -1,19 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isSummary = void 0;
-function isSummary(raw) {
-    if (typeof raw !== 'object') {
-        return false;
-    }
-    if (raw === null) {
-        return false;
-    }
-    var _a = raw, preview = _a.preview, overview = _a.overview;
-    if (!(typeof preview === 'string' && typeof overview === 'string')) {
-        return false;
-    }
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    var _ = { preview: preview, overview: overview };
-    return true;
-}
-exports.isSummary = isSummary;

package/dist/core/task/task.d.ts

@@ -1,10 +0,0 @@
-import { DisplayInformation } from './display_information';
-import { TrainingInformation } from './training_information';
-export declare type TaskID = string;
-export declare function isTaskID(obj: unknown): obj is TaskID;
-export declare function isTask(raw: unknown): raw is Task;
-export interface Task {
-    taskID: TaskID;
-    displayInformation: DisplayInformation;
-    trainingInformation: TrainingInformation;
-}

package/dist/core/task/task.js

@@ -1,31 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isTask = exports.isTaskID = void 0;
-var display_information_1 = require("./display_information");
-var training_information_1 = require("./training_information");
-function isTaskID(obj) {
-    return typeof obj === 'string';
-}
-exports.isTaskID = isTaskID;
-function isTask(raw) {
-    if (typeof raw !== 'object') {
-        return false;
-    }
-    if (raw === null) {
-        return false;
-    }
-    var _a = raw, taskID = _a.taskID, displayInformation = _a.displayInformation, trainingInformation = _a.trainingInformation;
-    if (typeof taskID !== 'string') {
-        return false;
-    }
-    if (!(0, display_information_1.isDisplayInformation)(displayInformation)) {
-        return false;
-    }
-    if (!(0, training_information_1.isTrainingInformation)(trainingInformation)) {
-        return false;
-    }
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    var _ = { taskID: taskID, displayInformation: displayInformation, trainingInformation: trainingInformation };
-    return true;
-}
-exports.isTask = isTask;

package/dist/core/task/training_information.d.ts

@@ -1,28 +0,0 @@
-import { Preprocessing } from '../dataset/data/preprocessing';
-import { ModelCompileData } from './model_compile_data';
-export declare function isTrainingInformation(raw: unknown): raw is TrainingInformation;
-export interface TrainingInformation {
-    modelID: string;
-    epochs: number;
-    roundDuration: number;
-    validationSplit: number;
-    batchSize: number;
-    preprocessingFunctions?: Preprocessing[];
-    modelCompileData: ModelCompileData;
-    dataType: string;
-    inputColumns?: string[];
-    outputColumns?: string[];
-    IMAGE_H?: number;
-    IMAGE_W?: number;
-    modelURL?: string;
-    LABEL_LIST?: string[];
-    learningRate?: number;
-    scheme: string;
-    noiseScale?: number;
-    clippingRadius?: number;
-    decentralizedSecure?: boolean;
-    byzantineRobustAggregator?: boolean;
-    tauPercentile?: number;
-    maxShareValue?: number;
-    minimumReadyPeers?: number;
-}

package/dist/core/task/training_information.js

@@ -1,66 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isTrainingInformation = void 0;
-var model_compile_data_1 = require("./model_compile_data");
-function isTrainingInformation(raw) {
-    if (typeof raw !== 'object') {
-        return false;
-    }
-    if (raw === null) {
-        return false;
-    }
-    var _a = raw, dataType = _a.dataType, scheme = _a.scheme, epochs = _a.epochs, 
-    // roundDuration,
-    validationSplit = _a.validationSplit, batchSize = _a.batchSize, modelCompileData = _a.modelCompileData, modelID = _a.modelID, preprocessingFunctions = _a.preprocessingFunctions, inputColumns = _a.inputColumns, outputColumns = _a.outputColumns, IMAGE_H = _a.IMAGE_H, IMAGE_W = _a.IMAGE_W, roundDuration = _a.roundDuration, modelURL = _a.modelURL, learningRate = _a.learningRate, decentralizedSecure = _a.decentralizedSecure, maxShareValue = _a.maxShareValue, minimumReadyPeers = _a.minimumReadyPeers, LABEL_LIST = _a.LABEL_LIST, noiseScale = _a.noiseScale, clippingRadius = _a.clippingRadius;
-    if (typeof dataType !== 'string' ||
-        typeof modelID !== 'string' ||
-        typeof epochs !== 'number' ||
-        typeof batchSize !== 'number' ||
-        typeof roundDuration !== 'number' ||
-        typeof validationSplit !== 'number' ||
-        (modelURL !== undefined && typeof modelURL !== 'string') ||
-        (noiseScale !== undefined && typeof noiseScale !== 'number') ||
-        (clippingRadius !== undefined && typeof clippingRadius !== 'number') ||
-        (learningRate !== undefined && typeof learningRate !== 'number') ||
-        (decentralizedSecure !== undefined && typeof decentralizedSecure !== 'boolean') ||
-        (maxShareValue !== undefined && typeof maxShareValue !== 'number') ||
-        (minimumReadyPeers !== undefined && typeof minimumReadyPeers !== 'number')) {
-        return false;
-    }
-    // interdepences on data type
-    switch (dataType) {
-        case 'image':
-            if (typeof IMAGE_H !== 'number' || typeof IMAGE_W !== 'number') {
-                return false;
-            }
-            break;
-        case 'tabular':
-            if (!(Array.isArray(inputColumns) && inputColumns.every(function (e) { return typeof e === 'string'; }))) {
-                return false;
-            }
-            if (!(Array.isArray(outputColumns) && outputColumns.every(function (e) { return typeof e === 'string'; }))) {
-                return false;
-            }
-            break;
-    }
-    // interdepences on scheme
-    switch (scheme) {
-        case 'decentralized':
-            break;
-        case 'federated':
-            break;
-        case 'local':
-            break;
-    }
-    if (!(0, model_compile_data_1.isModelCompileData)(modelCompileData)) {
-        return false;
-    }
-    if (LABEL_LIST !== undefined && !(Array.isArray(LABEL_LIST) && LABEL_LIST.every(function (e) { return typeof e === 'string'; }))) {
-        return false;
-    }
-    if (preprocessingFunctions !== undefined && !(Array.isArray(preprocessingFunctions) && preprocessingFunctions.every(function (e) { return typeof e === 'string'; }))) {
-        return false;
-    }
-    return true;
-}
-exports.isTrainingInformation = isTrainingInformation;

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

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

package/dist/core/tasks/cifar10.js

@@ -1,65 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.model = exports.task = void 0;
-var tslib_1 = require("tslib");
-var __1 = require("..");
-exports.task = {
-    taskID: 'cifar10',
-    displayInformation: {
-        taskTitle: 'CIFAR10',
-        summary: {
-            preview: 'In this challenge, we ask you to classify images into categories based on the objects shown on the image.',
-            overview: 'The CIFAR-10 dataset is a collection of images that are commonly used to train machine learning and computer vision algorithms. It is one of the most widely used datasets for machine learning research.'
-        },
-        limitations: 'The training data is limited to small images of size 32x32.',
-        tradeoffs: 'Training success strongly depends on label distribution',
-        dataFormatInformation: 'Images should be of .png format and of size 32x32. <br> The label file should be .csv, where each row contains a file_name, class.  <br> <br> e.g. if you have images: 0.png (of a frog) and 1.png (of a car) <br> labels.csv contains: (Note that no header is needed)<br> 0.png, frog <br> 1.png, car',
-        dataExampleText: 'Below you can find 10 random examples from each of the 10 classes in the dataset.',
-        dataExampleImage: 'https://storage.googleapis.com/deai-313515.appspot.com/example_training_data/cifar10-example.png'
-    },
-    trainingInformation: {
-        modelID: 'cifar10-model',
-        epochs: 10,
-        roundDuration: 10,
-        validationSplit: 0.2,
-        batchSize: 10,
-        modelCompileData: {
-            optimizer: 'sgd',
-            loss: 'categoricalCrossentropy',
-            metrics: ['accuracy']
-        },
-        dataType: 'image',
-        IMAGE_H: 32,
-        IMAGE_W: 32,
-        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() {
-    return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-        var mobilenet, x, predictions;
-        return (0, tslib_1.__generator)(this, function (_a) {
-            switch (_a.label) {
-                case 0: return [4 /*yield*/, __1.tf.loadLayersModel('https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_0.25_224/model.json')];
-                case 1:
-                    mobilenet = _a.sent();
-                    x = mobilenet.getLayer('global_average_pooling2d_1');
-                    predictions = __1.tf.layers
-                        .dense({ units: 10, activation: 'softmax', name: 'denseModified' })
-                        .apply(x.output);
-                    return [2 /*return*/, __1.tf.model({
-                            inputs: mobilenet.input,
-                            outputs: predictions,
-                            name: 'modelModified'
-                        })];
-            }
-        });
-    });
-}
-exports.model = model;

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

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

package/dist/core/tasks/geotags.js

@@ -1,67 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.model = exports.task = void 0;
-var tslib_1 = require("tslib");
-var __1 = require("..");
-var immutable_1 = require("immutable");
-exports.task = {
-    taskID: 'geotags',
-    displayInformation: {
-        taskTitle: 'GeoTags',
-        summary: {
-            preview: 'In this challenge, we predict the geo-location of a photo given its pixels in terms of a cell number of a grid built on top of Switzerland',
-            overview: 'The geotags dataset is a collection of images with geo-location information used to train a machine learning algorithm to predict the location of a photo given its pixels.'
-        },
-        limitations: 'The training data is limited to images of size 224x224.',
-        tradeoffs: 'Training success strongly depends on label distribution',
-        dataFormatInformation: 'Images should be of .png format and of size 224x224. <br> The label file should be .csv, where each row contains a file_name, class.  The class is the cell number of a the given grid of Switzerland. '
-    },
-    trainingInformation: {
-        modelID: 'geotags-model',
-        epochs: 10,
-        roundDuration: 10,
-        validationSplit: 0.2,
-        batchSize: 10,
-        modelCompileData: {
-            optimizer: 'adam',
-            loss: 'categoricalCrossentropy',
-            metrics: ['accuracy']
-        },
-        dataType: 'image',
-        IMAGE_H: 224,
-        IMAGE_W: 224,
-        preprocessingFunctions: [__1.data.ImagePreprocessing.Resize],
-        LABEL_LIST: (0, immutable_1.Range)(0, 140).map(String).toArray(),
-        scheme: 'Federated',
-        noiseScale: undefined,
-        clippingRadius: 20,
-        decentralizedSecure: true,
-        minimumReadyPeers: 3,
-        maxShareValue: 100
-    }
-};
-function model(_) {
-    if (_ === void 0) { _ = ''; }
-    return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-        var pretrainedModel, numLayers, model;
-        return (0, tslib_1.__generator)(this, function (_a) {
-            switch (_a.label) {
-                case 0: return [4 /*yield*/, __1.tf.loadLayersModel('https://storage.googleapis.com/epfl-disco-models/geotags/v2/model.json')];
-                case 1:
-                    pretrainedModel = _a.sent();
-                    numLayers = pretrainedModel.layers.length;
-                    pretrainedModel.layers.forEach(function (layer) { layer.trainable = false; });
-                    pretrainedModel.layers[numLayers - 1].trainable = true;
-                    model = __1.tf.sequential({
-                        layers: [
-                            __1.tf.layers.inputLayer({ inputShape: [224, 224, 3] }),
-                            __1.tf.layers.rescaling({ scale: 1 / 127.5, offset: -1 }),
-                            pretrainedModel
-                        ]
-                    });
-                    return [2 /*return*/, model];
-            }
-        });
-    });
-}
-exports.model = model;

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

@@ -1,6 +0,0 @@
-export * as cifar10 from './cifar10';
-export * as lus_covid from './lus_covid';
-export * as mnist from './mnist';
-export * as titanic from './titanic';
-export * as simple_face from './simple_face';
-export * as geotags from './geotags';