#include <iostream>
#include <nnann.h>
#include <nnann_function_step.h>
using namespace std;

void CreateTrainingSet(string fname);


//-------------
// main
//-------------
int main(int narg, char *argv[])
{
	// Create an ANN that we can train
	nnann ann;
	ann.AddLayer(2);
	ann.AddLayer(4);
	ann.AddLayer(1, new nnann_function_step());
	//ann.AddLayer(1, new nnann_function_step());
	
	// Train the ANN and save it to a file
	unsigned int epochs = ann.TrainFromFile("xor.train");
	//ann.SaveAs("xor.net");
	
	cout<<"Training complete after "<<epochs<<" passes through training set."<<endl;

	return 0;
}