tensorset/lib/Tensorset.js

TensorflowJS Dataset parsing and stringifying

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
const tf = __importStar(require("@tensorflow/tfjs"));
/**
* Tensorset is a class used similarly to JSON for stringifying
* and parsing Google's KNN Classifier datasets
* @author Mark Auger
* @version 1.2.9
* @social Follow me on GitHub at https://github.com/swimauger
* @license
* Copyright (c) 2020 Mark Auger
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
* @homepage https://github.com/swimauger/tensorset#readme
* @issues https://github.com/swimauger/tensorset/issues
**/
class Tensorset {
    /**
    * Parse stringified version of Tensorset generated by `Tensorset.stringify`
    * @param {string} data - Stringified tensorset file generated with assistance of `Tensorset.stringify`
    * @example
    * const dataset = Tensorset.parse(data);
    * @returns { [label: string]: Tensor2D } Returns KNN Classifier readable dataset
    **/
    static parse(data) {
        return JSON.parse(data).reduce((dataset, { label, values, shape }) => {
            return Object.assign(Object.assign({}, dataset), { [label]: tf.tensor(values, shape) });
        }, {});
    }
    /**
    * Stringify Google's KNN Classifier `getClassifierDataset` method
    * @param {any} dataset - JSON parsed version of `Tensorset.parse`
    * @example
    * const data = await Tensorset.stringify(dataset);
    * @returns {Promise<string>} Returns stringified tensorset of KNN Classifier dataset
    **/
    static async stringify(dataset) {
        return JSON.stringify(await Promise.all(Object.entries(dataset).map(async ([label, value]) => {
            return {
                label,
                values: Array.from(await value.data()),
                shape: value.shape,
            };
        })));
    }
}
Tensorset.default = Tensorset;
module.exports = Tensorset;
//# sourceMappingURL=Tensorset.js.map