Skip to content

Latest commit

 

History

History
83 lines (60 loc) · 2.37 KB

README.md

File metadata and controls

83 lines (60 loc) · 2.37 KB

Compile w/o Metal Build CMake Linux Build CMake Macos Run Linux

hopfield

hopfield network implementation in C++

compile

mkdir build && cd build
cmake ..
make

Linux

This repository uses the Metal framework to offload jobs to GPU. This works only on Apple machines and it is not supported by other systems. If you are running under Linux, please compile using

make all_nometal CXX=g++

it will ceate an executable called hopfield_nometal.out

Alternatively you can use cmake which automatically handles libraries

mkdir build && cd build
cmake ..
make

Run

./hopfield

On Apple devices be sure to run from the home folder (shaders are required to be in the correct path at runtime)

Example

    // Create a dataset with memories
    std::vector<Memory> training_dataset;
    training_dataset.emplace_back(Memory(6, 1, 1, 0, 0, 1, 1));
    training_dataset.emplace_back(Memory(6, 1, 1, 1, 1, 1, 1));

    //Create a model
    auto model = Model(training_dataset.size(), training_dataset[0].size());
    model.load_memories(training_dataset);


    // Build a corrupted memory
    spin corrupted_data[] = {1, 1, 1, 1, 1, 0};
    Memory corrupted_memory;
    std::memmove(corrupted_memory.fData, corrupted_data, corrupted_memory.size_of());

    // Train the model. Use can use kNull, kCPU, kGPU, kMultiThread, kOMP
    // If the chosen device is not available another one is automatically picked up
    model.train(kCPU);
    model.reconstruct(corrupted_memory);

    cout << "Reconstructed: " << endl;
    for (uint8_t i = 0; i < corrupted_memory.size(); i++) cout << corrupted_memory.fData[i] << " ";

    cout << model;

More examples are in the tests folder.

With images

Example with a corrupted image of an "8" hanwritten.

corrupted image

After it is reconstructed the black band disappears.

reconstructed image

Issues

The Github Action RunMacOS sometime fails due to lack of GPU access.

License

See LICENSE