encog/tests/neural/training/levenbergMarquardt.specs.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.

describe('Levenberg Marquardt Training', function () {
    const Encog = require(PATHS.BASE);
    Encog.Log.options.logLevel = 'silent';

    const LevenbergMarquardt = Encog.Training.LevenbergMarquardt;
    const NetworkUtil = Encog.Utils.Network;
    const Datasets = Encog.Utils.Datasets;

    test('Iris Flower Dataset', function () {
        // train the neural network
        const irisDataset = Datasets.getNormalizedIrisDataSet();
        const network = NetworkUtil.createIrisNetwork();

        const train = new LevenbergMarquardt(network, irisDataset.train.input, irisDataset.train.output);

        NetworkUtil.trainNetwork(train, {minError: 0.01, minIterations: 5});
        const accuracy = NetworkUtil.validateNetwork(network, irisDataset.test.input, irisDataset.test.output);

        expect(accuracy >= 0).toBeTruthy(); //TODO: Fix me
    });
});