encog/src/utils/file.js

Encog is a NodeJs ES6 framework based on the Encog Machine Learning Framework by Jeff Heaton, plus some the of basic data manipulation helpers.

const fs = require("fs");
const Networks = require('require-all')(PATHS.NETWORKS);
const EncogError = require(PATHS.ERROR_HANDLING + 'encog');
const _ = require('lodash');

class FileUtils {
    constructor() {

    }

    /**
     * @param network {BasicNetwork | HopfieldNetwork}
     * @param filename {String}
     */
    static saveNetwork(network, filename) {
        const networkData = JSON.stringify(network);

        fs.writeFileSync(filename, networkData);
    }

    /**
     * @param filename {String}
     * @returns {BasicNetwork | HopfieldNetwork}
     */
    static loadNetwork(filename) {
        const fileContent = fs.readFileSync(filename);
        console.log(fileContent);
        const networkData = JSON.parse(fileContent);
        let newBasicNetwork;
        const validNetworkTypes = ['BasicNetwork', 'HopfieldNetwork'];

        if (!networkData.type || validNetworkTypes.indexOf(networkData.type) === -1) {
            throw new EncogError('Not a valid network type');
        }

        const networkType = _.toLower(_.trim(networkData.type, 'Network'));

        newBasicNetwork = new Networks[networkType]();
        newBasicNetwork.fromJSON(networkData);

        return newBasicNetwork;
    }
}

module.exports = FileUtils;