jens
6ad74f4ee7
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@560 b431acfa-c32f-4a4a-93f1-934dc6c82436
43 lines
1.1 KiB
C++
43 lines
1.1 KiB
C++
#include <cstdio>
|
|
#include <cmath>
|
|
#include <armadillo>
|
|
#include "Rbm.hpp"
|
|
|
|
int main()
|
|
{
|
|
printf("Hallo, Welt!\n");
|
|
|
|
// Position of a particle // |
|
|
arma::vec Pos = {{0}, // | (0,1)
|
|
{1}}; // +---x-->
|
|
|
|
// Rotation matrix
|
|
double phi = -3.1416/2;
|
|
arma::mat RotM = {{+cos(phi), -sin(phi)},
|
|
{+sin(phi), +cos(phi)}};
|
|
|
|
Pos.print("Current position of the particle:");
|
|
std::cout << "Rotating the point " << phi*180/3.1416 << " deg" << std::endl;
|
|
|
|
Pos = RotM*Pos;
|
|
|
|
Pos.print("New position of the particle:"); // ^
|
|
// x (1,0)
|
|
// |
|
|
arma::mat Z = arma::eye<arma::mat>(4000,4000);
|
|
// Z.print("Z:");
|
|
printf("Z.n_rows = %d\n", (int)Z.n_rows);
|
|
printf("Z.n_cols = %d\n", (int)Z.n_cols);
|
|
printf("Z.n_elem = %d\n", (int)Z.n_elem);
|
|
// +------>
|
|
|
|
Rbm::Params params;
|
|
Rbm rbm(28*28, 64);
|
|
arma::mat batch = arma::randu(1000, 28*28);
|
|
rbm.train(batch, 1000, 100, params, nullptr);
|
|
|
|
// arma::mat v = arma::randu(28*28, 1);
|
|
// arma::mat h = rbm.toHidden(v);
|
|
// arma::mat r = rbm.toVisible(h);
|
|
return 0;
|
|
} |