test_hopfield.java

来自「Practical AI in Java 人工智能编程」· Java 代码 · 共 52 行

/**
 * Title:        <p>
 * Description:  <p>
 * Copyright:    Copyright (c) <p>
 * Company:      <p>
 * @author
 * @version 1.0
 */


public class Test_Hopfield {
    static float[] data [] = {
        {1, 1, 1, -1, -1, -1, -1, -1, -1, -1},
        {-1, -1, -1, 1, 1, 1, -1, -1, -1, -1},
        {-1, -1, -1, -1, -1, -1, -1, 1, 1, 1}
    };
    static Hopfield test;

    public static void main(String[] args) {
        test = new Hopfield(10);
        test.addTrainingData(data[0]);
        test.addTrainingData(data[1]);
        test.addTrainingData(data[2]);
        test.train();
        helper(test, "pattern 0", data[0]);
        helper(test, "pattern 1", data[1]);
        helper(test, "pattern 2", data[2]);
    }

    private static void helper(Hopfield test, String s, float[] test_data) {
        float[] dd = new float[10];
        for (int i = 0; i < 10; i++) {
            dd[i] = test_data[i];
        }
        int index = (int) (9.0f * (float) Math.random());
        if (dd[index] < 0.0f)
            dd[index] = 1.0f;
        else
            dd[index] = -1.0f;
        float[] rr = test.recall(dd, 5);
        System.out.println(s);
        for (int i = 0; i < 10; i++) System.out.print(pp(rr[i]) + " ");
        System.out.println();
    }

    private static int pp(float x) {
        if (x > 0.1f) return 1;
        return 0;
    }
}