encog/tests/neural/networks/hopfield.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('Hopfield Network', function () {
    const Encog = require(PATHS.BASE);
    Encog.Log.options.logLevel = 'silent';

    const HopfieldNetwork = Encog.Networks.Hopfield;
    const HopfieldPattern = Encog.Patterns.Hopfield;
    let network;
    let pattern;

    beforeEach(function () {
        pattern = new HopfieldPattern();
        pattern.setInputLayer(8);
        network = pattern.generate();
        network.addPattern([0, 0, 1, 0, 1, 1, 0, 1]);
    });

    describe('toJSON and fromJSON methods', function () {
         test('should save and restore a network', function () {
            const networkJSON = network.toJSON();
            expect(networkJSON.type).toBe('HopfieldNetwork');
            expect(networkJSON.neuronCount).toBe(8);
            expect(networkJSON.currentState).toEqual([-1, -1, -1, -1, -1, -1, -1, -1]);
            expect(networkJSON.weights.length).toBe(64);

            const newBasicNetwork = new HopfieldNetwork(2);
            newBasicNetwork.fromJSON(networkJSON);

            expect(newBasicNetwork.weights).toEqual(network.weights);
            expect(newBasicNetwork.neuronCount).toBe(network.neuronCount);
        });
    });
});