@epfml/discojs 2.1.1 → 2.1.2-p20240506085037.0

package/dist/core/client/federated/client.js

@@ -1,273 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Client = void 0;
-var tslib_1 = require("tslib");
-var uuid_1 = require("uuid");
-var __1 = require("../..");
-var base_1 = require("../base");
-var messages = (0, tslib_1.__importStar)(require("./messages"));
-var messages_1 = require("../messages");
-var nodeUrl = (0, tslib_1.__importStar)(require("url"));
-var event_connection_1 = require("../event_connection");
-var utils_1 = require("../utils");
-/**
- * Class that deals with communication with the centralized server when training
- * a specific task in the federated setting.
- */
-var Client = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(Client, _super);
-    function Client() {
-        var _this = _super !== null && _super.apply(this, arguments) || this;
-        _this.clientID = (0, uuid_1.v4)();
-        _this.round = 0;
-        return _this;
-    }
-    Object.defineProperty(Client.prototype, "server", {
-        get: function () {
-            if (this._server === undefined) {
-                throw new Error('server undefined, not connected');
-            }
-            return this._server;
-        },
-        enumerable: false,
-        configurable: true
-    });
-    // It opens a new WebSocket connection and listens to new messages over the channel
-    Client.prototype.connectServer = function (url) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var server;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, event_connection_1.WebSocketServer.connect(url, messages.isMessageFederated, messages.isMessageFederated)];
-                    case 1:
-                        server = _a.sent();
-                        return [2 /*return*/, server];
-                }
-            });
-        });
-    };
-    /**
-     * Initialize the connection to the server. TODO: In the case of FeAI,
-     * should return the current server-side round for the task.
-     */
-    Client.prototype.connect = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var URL, serverURL, _a, msg;
-            return (0, tslib_1.__generator)(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        URL = typeof window !== 'undefined' ? window.URL : nodeUrl.URL;
-                        serverURL = new URL('', this.url.href);
-                        switch (this.url.protocol) {
-                            case 'http:':
-                                serverURL.protocol = 'ws:';
-                                break;
-                            case 'https:':
-                                serverURL.protocol = 'wss:';
-                                break;
-                            default:
-                                throw new Error("unknown protocol: " + this.url.protocol);
-                        }
-                        serverURL.pathname += "feai/" + this.task.taskID + "/" + this.clientID;
-                        _a = this;
-                        return [4 /*yield*/, this.connectServer(serverURL)];
-                    case 1:
-                        _a._server = _b.sent();
-                        msg = {
-                            type: messages_1.type.clientConnected
-                        };
-                        this.server.send(msg);
-                        return [4 /*yield*/, (0, event_connection_1.waitMessageWithTimeout)(this.server, messages_1.type.clientConnected, utils_1.MAX_WAIT_PER_ROUND)];
-                    case 2:
-                        _b.sent();
-                        this.connected = true;
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    /**
-     * Disconnection process when user quits the task.
-     */
-    Client.prototype.disconnect = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                this.server.disconnect();
-                this._server = undefined;
-                this.connected = false;
-                return [2 /*return*/];
-            });
-        });
-    };
-    // It sends a message to the server
-    Client.prototype.sendMessage = function (msg) {
-        var _a;
-        (_a = this.server) === null || _a === void 0 ? void 0 : _a.send(msg);
-    };
-    // It sends weights to the server
-    Client.prototype.postWeightsToServer = function (weights) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var msg;
-            var _a;
-            return (0, tslib_1.__generator)(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        _a = {
-                            type: messages_1.type.postWeightsToServer
-                        };
-                        return [4 /*yield*/, __1.serialization.weights.encode(weights)];
-                    case 1:
-                        msg = (_a.weights = _b.sent(),
-                            _a.round = this.round,
-                            _a);
-                        this.sendMessage(msg);
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    // It retrieves the last server round and weights, but return only the server round
-    Client.prototype.getLatestServerRound = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var msg, received;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        this.serverRound = undefined;
-                        this.serverWeights = undefined;
-                        msg = {
-                            type: messages_1.type.latestServerRound
-                        };
-                        this.sendMessage(msg);
-                        return [4 /*yield*/, (0, event_connection_1.waitMessageWithTimeout)(this.server, messages_1.type.latestServerRound, utils_1.MAX_WAIT_PER_ROUND)];
-                    case 1:
-                        received = _a.sent();
-                        this.serverRound = received.round;
-                        this.serverWeights = __1.serialization.weights.decode(received.weights);
-                        return [2 /*return*/, this.serverRound];
-                }
-            });
-        });
-    };
-    // It retrieves the last server round and weights, but return only the server weights
-    Client.prototype.pullRoundAndFetchWeights = function () {
-        var _a;
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_b) {
-                switch (_b.label) {
-                    case 0: 
-                    // get server round of latest model
-                    return [4 /*yield*/, this.getLatestServerRound()];
-                    case 1:
-                        // get server round of latest model
-                        _b.sent();
-                        if (this.round < ((_a = this.serverRound) !== null && _a !== void 0 ? _a : 0)) {
-                            // Update the local round to match the server's
-                            this.round = this.serverRound;
-                            return [2 /*return*/, this.serverWeights];
-                        }
-                        else {
-                            return [2 /*return*/, undefined];
-                        }
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    // It pulls statistics from the server
-    Client.prototype.pullServerStatistics = function (trainingInformant) {
-        var _a;
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var msg, received;
-            return (0, tslib_1.__generator)(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        this.receivedStatistics = undefined;
-                        msg = {
-                            type: messages_1.type.pullServerStatistics
-                        };
-                        this.sendMessage(msg);
-                        return [4 /*yield*/, (0, event_connection_1.waitMessageWithTimeout)(this.server, messages_1.type.pullServerStatistics, utils_1.MAX_WAIT_PER_ROUND)];
-                    case 1:
-                        received = _b.sent();
-                        this.receivedStatistics = received.statistics;
-                        trainingInformant.update((_a = this.receivedStatistics) !== null && _a !== void 0 ? _a : {});
-                        return [2 /*return*/];
-                }
-            });
-        });
-    };
-    // It posts a new metadata value to the server
-    Client.prototype.postMetadata = function (metadataID, metadata) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var msg;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                msg = {
-                    type: messages_1.type.postMetadata,
-                    taskId: this.task.taskID,
-                    clientId: this.clientID,
-                    round: this.round,
-                    metadataId: metadataID,
-                    metadata: metadata
-                };
-                this.sendMessage(msg);
-                return [2 /*return*/];
-            });
-        });
-    };
-    // It gets a metadata map from the server
-    Client.prototype.getMetadataMap = function (metadataId) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var msg, received;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        this.metadataMap = undefined;
-                        msg = {
-                            type: messages_1.type.getMetadataMap,
-                            taskId: this.task.taskID,
-                            clientId: this.clientID,
-                            round: this.round,
-                            metadataId: metadataId
-                        };
-                        this.sendMessage(msg);
-                        return [4 /*yield*/, (0, event_connection_1.waitMessageWithTimeout)(this.server, messages_1.type.getMetadataMap, utils_1.MAX_WAIT_PER_ROUND)];
-                    case 1:
-                        received = _a.sent();
-                        if (received.metadataMap !== undefined) {
-                            this.metadataMap = new Map(received.metadataMap);
-                        }
-                        return [2 /*return*/, this.metadataMap];
-                }
-            });
-        });
-    };
-    Client.prototype.onRoundEndCommunication = function (updatedWeights, staleWeights, _, trainingInformant) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var noisyWeights, serverWeights;
-            return (0, tslib_1.__generator)(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        noisyWeights = __1.privacy.addDifferentialPrivacy(updatedWeights, staleWeights, this.task);
-                        return [4 /*yield*/, this.postWeightsToServer(noisyWeights)];
-                    case 1:
-                        _a.sent();
-                        return [4 /*yield*/, this.pullServerStatistics(trainingInformant)];
-                    case 2:
-                        _a.sent();
-                        return [4 /*yield*/, this.pullRoundAndFetchWeights()];
-                    case 3:
-                        serverWeights = _a.sent();
-                        return [2 /*return*/, serverWeights !== null && serverWeights !== void 0 ? serverWeights : staleWeights];
-                }
-            });
-        });
-    };
-    Client.prototype.onTrainEndCommunication = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () { return (0, tslib_1.__generator)(this, function (_a) {
-            return [2 /*return*/];
-        }); });
-    };
-    return Client;
-}(base_1.Base));
-exports.Client = Client;

package/dist/core/client/federated/index.d.ts

@@ -1,2 +0,0 @@
-export { Client } from './client';
-export * as messages from './messages';

package/dist/core/client/federated/index.js

@@ -1,7 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.messages = exports.Client = void 0;
-var tslib_1 = require("tslib");
-var client_1 = require("./client");
-Object.defineProperty(exports, "Client", { enumerable: true, get: function () { return client_1.Client; } });
-exports.messages = (0, tslib_1.__importStar)(require("./messages"));

package/dist/core/client/federated/messages.d.ts

@@ -1,38 +0,0 @@
-import { MetadataID } from '../..';
-import { weights } from '../../serialization';
-import { type } from '../messages';
-export declare type MessageFederated = postWeightsToServer | latestServerRound | pullServerStatistics | postMetadata | getMetadataMap | messageGeneral;
-export interface messageGeneral {
-    type: type;
-}
-export interface postWeightsToServer {
-    type: type.postWeightsToServer;
-    weights: weights.Encoded;
-    round: number;
-}
-export interface latestServerRound {
-    type: type.latestServerRound;
-    weights: weights.Encoded;
-    round: number;
-}
-export interface pullServerStatistics {
-    type: type.pullServerStatistics;
-    statistics: Record<string, number>;
-}
-export interface postMetadata {
-    type: type.postMetadata;
-    clientId: string;
-    taskId: string;
-    round: number;
-    metadataId: string;
-    metadata: string;
-}
-export interface getMetadataMap {
-    type: type.getMetadataMap;
-    clientId: string;
-    taskId: string;
-    round: number;
-    metadataId: MetadataID;
-    metadataMap?: Array<[string, string | undefined]>;
-}
-export declare function isMessageFederated(o: unknown): o is MessageFederated;

package/dist/core/client/federated/messages.js

@@ -1,25 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.isMessageFederated = void 0;
-var messages_1 = require("../messages");
-function isMessageFederated(o) {
-    if (!(0, messages_1.hasMessageType)(o)) {
-        return false;
-    }
-    switch (o.type) {
-        case messages_1.type.clientConnected:
-            return true;
-        case messages_1.type.postWeightsToServer:
-            return true;
-        case messages_1.type.latestServerRound:
-            return true;
-        case messages_1.type.pullServerStatistics:
-            return true;
-        case messages_1.type.postMetadata:
-            return true;
-        case messages_1.type.getMetadataMap:
-            return true;
-    }
-    return false;
-}
-exports.isMessageFederated = isMessageFederated;

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

@@ -1,5 +0,0 @@
-export { Base } from './base';
-export * as decentralized from './decentralized';
-export * as federated from './federated';
-export * as messages from './messages';
-export { Local } from './local';

package/dist/core/client/index.js

@@ -1,11 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Local = exports.messages = exports.federated = exports.decentralized = exports.Base = void 0;
-var tslib_1 = require("tslib");
-var base_1 = require("./base");
-Object.defineProperty(exports, "Base", { enumerable: true, get: function () { return base_1.Base; } });
-exports.decentralized = (0, tslib_1.__importStar)(require("./decentralized"));
-exports.federated = (0, tslib_1.__importStar)(require("./federated"));
-exports.messages = (0, tslib_1.__importStar)(require("./messages"));
-var local_1 = require("./local");
-Object.defineProperty(exports, "Local", { enumerable: true, get: function () { return local_1.Local; } });

package/dist/core/client/local.d.ts

@@ -1,8 +0,0 @@
-import { WeightsContainer } from '..';
-import { Base } from './base';
-export declare class Local extends Base {
-    connect(): Promise<void>;
-    disconnect(): Promise<void>;
-    onRoundEndCommunication(_: WeightsContainer): Promise<WeightsContainer>;
-    onTrainEndCommunication(): Promise<void>;
-}

package/dist/core/client/local.js

@@ -1,36 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Local = void 0;
-var tslib_1 = require("tslib");
-var base_1 = require("./base");
-// does pretty much nothing
-var Local = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(Local, _super);
-    function Local() {
-        return _super !== null && _super.apply(this, arguments) || this;
-    }
-    Local.prototype.connect = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () { return (0, tslib_1.__generator)(this, function (_a) {
-            return [2 /*return*/];
-        }); });
-    };
-    Local.prototype.disconnect = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () { return (0, tslib_1.__generator)(this, function (_a) {
-            return [2 /*return*/];
-        }); });
-    };
-    Local.prototype.onRoundEndCommunication = function (_) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                return [2 /*return*/, _];
-            });
-        });
-    };
-    Local.prototype.onTrainEndCommunication = function () {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () { return (0, tslib_1.__generator)(this, function (_a) {
-            return [2 /*return*/];
-        }); });
-    };
-    return Local;
-}(base_1.Base));
-exports.Local = Local;

package/dist/core/client/messages.d.ts

@@ -1,28 +0,0 @@
-import * as decentralized from './decentralized/messages';
-import * as federated from './federated/messages';
-export declare enum type {
-    clientConnected = 0,
-    PeerID = 1,
-    SignalForPeer = 2,
-    PeerIsReady = 3,
-    PeersForRound = 4,
-    Weights = 5,
-    Shares = 6,
-    PartialSums = 7,
-    postWeightsToServer = 8,
-    postMetadata = 9,
-    getMetadataMap = 10,
-    latestServerRound = 11,
-    pullRoundAndFetchWeights = 12,
-    pullServerStatistics = 13
-}
-export interface clientConnected {
-    type: type.clientConnected;
-}
-export declare type Message = decentralized.MessageFromServer | decentralized.MessageToServer | decentralized.PeerMessage | federated.MessageFederated;
-export declare type NarrowMessage<D> = Extract<Message, {
-    type: D;
-}>;
-export declare function hasMessageType(raw: unknown): raw is {
-    type: type;
-} & Record<string, unknown>;

package/dist/core/client/messages.js

@@ -1,33 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.hasMessageType = exports.type = void 0;
-var type;
-(function (type) {
-    type[type["clientConnected"] = 0] = "clientConnected";
-    // decentralized
-    type[type["PeerID"] = 1] = "PeerID";
-    type[type["SignalForPeer"] = 2] = "SignalForPeer";
-    type[type["PeerIsReady"] = 3] = "PeerIsReady";
-    type[type["PeersForRound"] = 4] = "PeersForRound";
-    type[type["Weights"] = 5] = "Weights";
-    type[type["Shares"] = 6] = "Shares";
-    type[type["PartialSums"] = 7] = "PartialSums";
-    // federated
-    type[type["postWeightsToServer"] = 8] = "postWeightsToServer";
-    type[type["postMetadata"] = 9] = "postMetadata";
-    type[type["getMetadataMap"] = 10] = "getMetadataMap";
-    type[type["latestServerRound"] = 11] = "latestServerRound";
-    type[type["pullRoundAndFetchWeights"] = 12] = "pullRoundAndFetchWeights";
-    type[type["pullServerStatistics"] = 13] = "pullServerStatistics";
-})(type = exports.type || (exports.type = {}));
-function hasMessageType(raw) {
-    if (typeof raw !== 'object' || raw === null) {
-        return false;
-    }
-    var o = raw;
-    if (!('type' in o && typeof o.type === 'number' && o.type in type)) {
-        return false;
-    }
-    return true;
-}
-exports.hasMessageType = hasMessageType;

package/dist/core/client/utils.d.ts

@@ -1,2 +0,0 @@
-export declare const MAX_WAIT_PER_ROUND = 10000;
-export declare function timeout(ms: number): Promise<never>;

package/dist/core/client/utils.js

@@ -1,19 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.timeout = exports.MAX_WAIT_PER_ROUND = void 0;
-var tslib_1 = require("tslib");
-// Time to wait for the others in milliseconds.
-exports.MAX_WAIT_PER_ROUND = 10000;
-function timeout(ms) {
-    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*/, new Promise(function (resolve, reject) {
-                        setTimeout(function () { return reject(new Error('timeout')); }, ms);
-                    })];
-                case 1: return [2 /*return*/, _a.sent()];
-            }
-        });
-    });
-}
-exports.timeout = timeout;

package/dist/core/dataset/data/data.d.ts

@@ -1,11 +0,0 @@
-import { Task } from '../..';
-import { Dataset } from '../dataset';
-export declare abstract class Data {
-    readonly dataset: Dataset;
-    readonly task: Task;
-    readonly size?: number | undefined;
-    protected constructor(dataset: Dataset, task: Task, size?: number | undefined);
-    static init(dataset: Dataset, task: Task, size?: number): Promise<Data>;
-    abstract batch(): Data;
-    abstract preprocess(): Data;
-}

package/dist/core/dataset/data/data.js

@@ -1,20 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Data = void 0;
-var tslib_1 = require("tslib");
-var Data = /** @class */ (function () {
-    function Data(dataset, task, size) {
-        this.dataset = dataset;
-        this.task = task;
-        this.size = size;
-    }
-    Data.init = function (dataset, task, size) {
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            return (0, tslib_1.__generator)(this, function (_a) {
-                throw new Error('abstract');
-            });
-        });
-    };
-    return Data;
-}());
-exports.Data = Data;

package/dist/core/dataset/data/data_split.d.ts

@@ -1,5 +0,0 @@
-import { Data } from './data';
-export interface DataSplit {
-    train: Data;
-    validation?: Data;
-}

package/dist/core/dataset/data/data_split.js

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/dist/core/dataset/data/image_data.d.ts

@@ -1,8 +0,0 @@
-import { Task } from '../..';
-import { Dataset } from '../dataset';
-import { Data } from './data';
-export declare class ImageData extends Data {
-    static init(dataset: Dataset, task: Task, size?: number): Promise<Data>;
-    batch(): Data;
-    preprocess(): Data;
-}

package/dist/core/dataset/data/image_data.js

@@ -1,64 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImageData = void 0;
-var tslib_1 = require("tslib");
-var preprocessing_1 = require("./preprocessing");
-var data_1 = require("./data");
-var ImageData = /** @class */ (function (_super) {
-    (0, tslib_1.__extends)(ImageData, _super);
-    function ImageData() {
-        return _super !== null && _super.apply(this, arguments) || this;
-    }
-    ImageData.init = function (dataset, task, size) {
-        var _a;
-        return (0, tslib_1.__awaiter)(this, void 0, void 0, function () {
-            var sample, shape, e_1;
-            return (0, tslib_1.__generator)(this, function (_b) {
-                switch (_b.label) {
-                    case 0:
-                        if (!!((_a = task.trainingInformation.preprocessingFunctions) === null || _a === void 0 ? void 0 : _a.includes(preprocessing_1.ImagePreprocessing.Resize))) return [3 /*break*/, 4];
-                        _b.label = 1;
-                    case 1:
-                        _b.trys.push([1, 3, , 4]);
-                        return [4 /*yield*/, dataset.take(1).toArray()];
-                    case 2:
-                        sample = (_b.sent())[0];
-                        // TODO: We suppose the presence of labels
-                        // TODO: Typing (discojs-node/src/dataset/data_loader/image_loader.spec.ts)
-                        if (!(typeof sample === 'object' && sample !== null)) {
-                            throw new Error();
-                        }
-                        shape = void 0;
-                        if ('xs' in sample && 'ys' in sample) {
-                            shape = sample.xs.shape;
-                        }
-                        else {
-                            shape = sample.shape;
-                        }
-                        if (!(shape[0] === task.trainingInformation.IMAGE_W &&
-                            shape[1] === task.trainingInformation.IMAGE_H)) {
-                            throw new Error();
-                        }
-                        return [3 /*break*/, 4];
-                    case 3:
-                        e_1 = _b.sent();
-                        throw new Error('Data input format is not compatible with the chosen task');
-                    case 4: return [2 /*return*/, new ImageData(dataset, task, size)];
-                }
-            });
-        });
-    };
-    ImageData.prototype.batch = function () {
-        var batchSize = this.task.trainingInformation.batchSize;
-        var newDataset = batchSize === undefined ? this.dataset : this.dataset.batch(batchSize);
-        return new ImageData(newDataset, this.task, this.size);
-    };
-    ImageData.prototype.preprocess = function () {
-        var newDataset = this.dataset;
-        var preprocessImage = (0, preprocessing_1.getPreprocessImage)(this.task);
-        newDataset = newDataset.map(function (x) { return preprocessImage(x); });
-        return new ImageData(newDataset, this.task, this.size);
-    };
-    return ImageData;
-}(data_1.Data));
-exports.ImageData = ImageData;

package/dist/core/dataset/data/index.d.ts

@@ -1,5 +0,0 @@
-export { DataSplit } from './data_split';
-export { Data } from './data';
-export { ImageData } from './image_data';
-export { TabularData } from './tabular_data';
-export { ImagePreprocessing } from './preprocessing';

package/dist/core/dataset/data/index.js

@@ -1,11 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImagePreprocessing = exports.TabularData = exports.ImageData = exports.Data = void 0;
-var data_1 = require("./data");
-Object.defineProperty(exports, "Data", { enumerable: true, get: function () { return data_1.Data; } });
-var image_data_1 = require("./image_data");
-Object.defineProperty(exports, "ImageData", { enumerable: true, get: function () { return image_data_1.ImageData; } });
-var tabular_data_1 = require("./tabular_data");
-Object.defineProperty(exports, "TabularData", { enumerable: true, get: function () { return tabular_data_1.TabularData; } });
-var preprocessing_1 = require("./preprocessing");
-Object.defineProperty(exports, "ImagePreprocessing", { enumerable: true, get: function () { return preprocessing_1.ImagePreprocessing; } });

package/dist/core/dataset/data/preprocessing.d.ts

@@ -1,13 +0,0 @@
-import { tf, Task } from '../..';
-declare type PreprocessImage = (image: tf.TensorContainer) => tf.TensorContainer;
-export declare type Preprocessing = ImagePreprocessing;
-export interface ImageTensorContainer extends tf.TensorContainerObject {
-    xs: tf.Tensor3D | tf.Tensor4D;
-    ys: tf.Tensor1D | number | undefined;
-}
-export declare enum ImagePreprocessing {
-    Normalize = "normalize",
-    Resize = "resize"
-}
-export declare function getPreprocessImage(task: Task): PreprocessImage;
-export {};