jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- numcontext is property of Layer

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@844 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-19 16:05:23 +00:00
parent 2f3c10095a
commit 6a05a88d15
7 changed files with 60 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/poet.cpp
+32 -27
View File
@@ -36,7 +36,7 @@ class RbmListener : public Rbm::IListener
#define CREATE_TRAINING 0
#define DO_TRAINING 0
#define DO_TRAINING 1
#define DO_FORWARD 1
int main()
@@ -51,39 +51,44 @@ int main()
RnnStack *stack = reinterpret_cast<RnnStack*>(StackCreator::fromFile(".", "poet_2v_5s"));
// Load weights
stack->loadWeights(".");
bool weight_loaded = stack->loadWeights(".");
// Load training
stack->loadTrainingBatch(".");
arma::mat t_vc = stack->trainingBatch();
bool do_training = not weight_loaded;
#if DO_TRAINING
RbmListener listener;
// Phase 10
stack->train(t_vc, &listener);
stack->saveWeights(".");
stack->save(".");
printf("Curr\n");
for (int i=0; i < t_vc.n_rows; i++)
{
arma::mat curr = stack->to_curr(t_vc.row(i));
char c = RnnTextHelper::idx2ch(curr.index_max());
putchar(c);
}
printf("\n");
printf("Next\n");
for (int i=0; i < t_vc.n_rows; i++)
{
arma::mat next = stack->to_next(t_vc.row(i));
char c = RnnTextHelper::idx2ch(next.index_max());
putchar(c);
}
printf("\n");
do_training = true;
#endif
if (do_training)
{
RbmListener listener;
// Phase 10
stack->train(t_vc, &listener);
stack->saveWeights(".");
stack->save(".");
printf("Curr\n");
for (int i=0; i < t_vc.n_rows; i++)
{
arma::mat curr = stack->to_curr(t_vc.row(i));
char c = RnnTextHelper::idx2ch(curr.index_max());
putchar(c);
}
printf("\n");
printf("Next\n");
for (int i=0; i < t_vc.n_rows; i++)
{
arma::mat next = stack->to_next(t_vc.row(i));
char c = RnnTextHelper::idx2ch(next.index_max());
putchar(c);
}
printf("\n");
}
#if DO_FORWARD
arma::mat state;