74 lines
No EOL
1.4 KiB
C++
74 lines
No EOL
1.4 KiB
C++
#include <iostream>
|
|
#include <exception>
|
|
#include <algorithm>
|
|
|
|
#include <assert.h>
|
|
|
|
#include "Net.h"
|
|
|
|
const double pi = std::acos(-1);
|
|
|
|
int main()
|
|
{
|
|
try
|
|
{
|
|
std::cout << "Neuro running" << std::endl;
|
|
|
|
Net myNet({ 2, 3, 1 });
|
|
|
|
size_t batchSize = 5000;
|
|
size_t batchIndex = 0;
|
|
double batchMaxError = 0.0;
|
|
double batchMeanError = 0.0;
|
|
|
|
size_t numIterations = 100000;
|
|
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)
|
|
{
|
|
std::cout << "Batch error (" << batchSize << " iterations, max/mean): ";
|
|
std::cout << std::abs(batchMaxError);
|
|
std::cout << " / ";
|
|
std::cout << std::abs(batchMeanError / batchSize);
|
|
std::cout << std::endl;
|
|
|
|
batchIndex = 0;
|
|
batchMaxError = 0.0;
|
|
batchMeanError = 0.0;
|
|
}
|
|
|
|
myNet.backProp(targetValues);
|
|
}
|
|
|
|
myNet.save("mynet.nnet");
|
|
Net copyNet("mynet.nnet");
|
|
}
|
|
catch (std::exception &ex)
|
|
{
|
|
std::cerr << "Error: " << ex.what() << std::endl;
|
|
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
} |