MachineIntelligenceCore:NeuralNets
|
Program for visualization of features of hebbian network trained on MNIST digits. : Tomasz Kornuta tkorn : May 16, 2017. ut@u s.ibm .comMore...
#include <boost/thread/thread.hpp>
#include <boost/bind.hpp>
#include <importers/MNISTMatrixImporter.hpp>
#include <logger/Log.hpp>
#include <logger/ConsoleOutput.hpp>
#include <application/ApplicationState.hpp>
#include <configuration/ParameterServer.hpp>
#include <opengl/visualization/WindowManager.hpp>
#include <opengl/visualization/WindowGrayscaleBatch.hpp>
#include <mlnn/HebbianNeuralNetwork.hpp>
#include <encoders/MatrixXfMatrixXfEncoder.hpp>
#include <encoders/UIntMatrixXfEncoder.hpp>
Go to the source code of this file.
Functions | |
void | batch_function (void) |
Function for batch sampling. More... | |
int | main (int argc, char *argv[]) |
Main program function. Runs two threads: main (for GLUT) and another one (for data processing). More... | |
Variables | |
WindowGrayscaleBatch< float > * | w_input |
Window for displaying the MNIST batch. More... | |
WindowGrayscaleBatch< float > * | w_reconstruction |
WindowGrayscaleBatch< float > * | w_weights1 |
Window for displaying the weights. More... | |
WindowGrayscaleBatch< float > * | w_weights2 |
mic::importers::MNISTMatrixImporter < float > * | importer |
MNIST importer. More... | |
HebbianNeuralNetwork< float > | neural_net |
Multi-layer neural network. More... | |
mic::encoders::MatrixXfMatrixXfEncoder * | mnist_encoder |
MNIST matrix encoder. More... | |
const size_t | patch_size = 28 |
Label 2 matrix encoder (1 hot). More... | |
const size_t | batch_size = 4 |
const size_t | output_units = 12 |
Program for visualization of features of hebbian network trained on MNIST digits. : Tomasz Kornuta tkorn : May 16, 2017. ut@u s.ibm .com
Copyright (C) tkornuta, IBM Corporation 2015-2019
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
Copyright (c) 2017, Tomasz Kornuta, IBM Corporation. All rights reserved.
Definition in file mnist_hebbian_features_visualization_test.cpp.
void batch_function | ( | void | ) |
Function for batch sampling.
Definition at line 76 of file mnist_hebbian_features_visualization_test.cpp.
References importer, mnist_encoder, neural_net, output_units, patch_size, w_input, w_reconstruction, and w_weights1.
Referenced by main().
int main | ( | int | argc, |
char * | argv[] | ||
) |
Main program function. Runs two threads: main (for GLUT) and another one (for data processing).
[in] | argc | Number of parameters (passed to glManaged). |
[in] | argv | List of parameters (passed to glManaged). |
Definition at line 151 of file mnist_hebbian_features_visualization_test.cpp.
References batch_function(), batch_size, importer, mnist_encoder, patch_size, w_input, w_reconstruction, and w_weights1.
const size_t batch_size = 4 |
Definition at line 69 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by main().
mic::importers::MNISTMatrixImporter<float>* importer |
MNIST importer.
Definition at line 59 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function(), and main().
mic::encoders::MatrixXfMatrixXfEncoder* mnist_encoder |
MNIST matrix encoder.
Definition at line 64 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function(), and main().
HebbianNeuralNetwork<float> neural_net |
Multi-layer neural network.
Definition at line 61 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function().
const size_t output_units = 12 |
Definition at line 70 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function().
const size_t patch_size = 28 |
Label 2 matrix encoder (1 hot).
Definition at line 68 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function(), and main().
WindowGrayscaleBatch<float>* w_input |
Window for displaying the MNIST batch.
Definition at line 52 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function(), and main().
WindowGrayscaleBatch<float>* w_reconstruction |
Definition at line 53 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function(), and main().
WindowGrayscaleBatch<float>* w_weights1 |
Window for displaying the weights.
Definition at line 55 of file mnist_hebbian_features_visualization_test.cpp.
Referenced by batch_function(), and main().
WindowGrayscaleBatch<float>* w_weights2 |
Definition at line 56 of file mnist_hebbian_features_visualization_test.cpp.