95 lines
2.5 KiB
C++
95 lines
2.5 KiB
C++
#include "netlearner.h"
|
|
#include "../../Net.h"
|
|
#include "trainingdataloader.h"
|
|
|
|
#include <QElapsedTimer>
|
|
|
|
void NetLearner::run()
|
|
{
|
|
try
|
|
{
|
|
QElapsedTimer timer;
|
|
|
|
TrainingDataLoader dataLoader;
|
|
dataLoader.addSamples("../NeuroUI/training data/mnist_train0.jpg", 0);
|
|
|
|
Net myNet;
|
|
try
|
|
{
|
|
myNet.load("mynet.nnet");
|
|
}
|
|
catch (...)
|
|
{
|
|
myNet.initialize({2, 3, 1});
|
|
}
|
|
|
|
size_t batchSize = 5000;
|
|
size_t batchIndex = 0;
|
|
double batchMaxError = 0.0;
|
|
double batchMeanError = 0.0;
|
|
|
|
timer.start();
|
|
|
|
size_t numIterations = 2000000;
|
|
for (size_t iteration = 0; iteration < numIterations; ++iteration)
|
|
{
|
|
std::vector<double> inputValues =
|
|
{
|
|
std::rand() / (double)RAND_MAX,
|
|
std::rand() / (double)RAND_MAX
|
|
};
|
|
|
|
std::vector<double> targetValues =
|
|
{
|
|
(inputValues[0] + inputValues[1]) / 2.0
|
|
};
|
|
|
|
myNet.feedForward(inputValues);
|
|
|
|
std::vector<double> outputValues = myNet.getOutput();
|
|
|
|
double error = outputValues[0] - targetValues[0];
|
|
|
|
batchMeanError += error;
|
|
batchMaxError = std::max<double>(batchMaxError, error);
|
|
|
|
if (batchIndex++ == batchSize)
|
|
{
|
|
QString logString;
|
|
|
|
logString.append("Batch error (");
|
|
logString.append(QString::number(batchSize));
|
|
logString.append(" iterations, max/mean): ");
|
|
logString.append(QString::number(std::abs(batchMaxError)));
|
|
logString.append(" / ");
|
|
logString.append(QString::number(std::abs(batchMeanError / batchSize)));
|
|
|
|
emit logMessage(logString);
|
|
emit currentNetError(batchMaxError);
|
|
emit progress((double)iteration / (double)numIterations);
|
|
|
|
batchIndex = 0;
|
|
batchMaxError = 0.0;
|
|
batchMeanError = 0.0;
|
|
}
|
|
|
|
myNet.backProp(targetValues);
|
|
}
|
|
|
|
QString timerLogString;
|
|
timerLogString.append("Elapsed time: ");
|
|
timerLogString.append(QString::number(timer.elapsed() / 1000.0));
|
|
timerLogString.append(" seconds");
|
|
|
|
emit logMessage(timerLogString);
|
|
|
|
myNet.save("mynet.nnet");
|
|
}
|
|
catch (std::exception &ex)
|
|
{
|
|
QString logString("Error: ");
|
|
logString.append(ex.what());
|
|
emit logMessage(logString);
|
|
}
|
|
}
|