void NextEpoch(vector<dna>& dnas)
{
size_t cnt = dnas.size();
#pragma omp parallel
#pragma omp for
for (short it = 0; it < cnt; ++it)
{
FANN::neural_net net;
net.create_standard(3, data.num_input_train_data(), dnas[it].numberOfHidden, data.num_output_train_data());
net.set_learning_rate(dnas[it].learningRate);
net.set_learning_momentum(dnas[it].momentum);
net.set_activation_steepness_hidden(1.0);
net.set_activation_steepness_output(1.0);
net.set_activation_function_hidden(FANN::LINEAR);
net.set_activation_function_output(FANN::LINEAR);
net.set_callback(print_fake, NULL);
net.set_error_log(NULL);
net.train_on_data(data, dnas[it].maxIteration,
-1, dnas[it].desiredError);
dnas[it].fitness = (1. - net.test_data(testdata)) ;
}
}
void PrintBest(vector<dna>& dnas, int gen)
{
cout << "Best gen : "<< gen << "----------" << endl;
for (int i = 0; i < gen; ++i)
{
cout << endl;
cout << "-********************-" << endl;
cout << "fitness" << dnas[i].fitness << endl;
cout << "learningRate" << dnas[i].learningRate << endl;
cout << "momentum" << dnas[i].momentum << endl;
cout << "numberOfHidden" << dnas[i].numberOfHidden << endl;
cout << "maxIteration" << dnas[i].maxIteration << endl;
cout << "desiredError" << dnas[i].desiredError<< endl;
cout << "-********************-" << endl;
cout << endl;
}
FANN::neural_net net;
net.create_standard(3, data.num_input_train_data(), dnas[0].numberOfHidden, data.num_output_train_data());
net.set_learning_rate(dnas[0].learningRate);
net.set_learning_momentum(dnas[0].momentum);
net.set_activation_steepness_hidden(1.0);
net.set_activation_steepness_output(1.0);
net.set_activation_function_hidden(FANN::SIGMOID);
net.set_activation_function_output(FANN::SIGMOID);
ofstream file("ErrorHistory/" + file + ".csv");
net.set_callback(print_callback, &file);
net.set_error_log(NULL);
net.train_on_data(data, dnas[0].maxIteration,
1, dnas[0].desiredError);
cout << "----------------------" << endl;
}
//Creates a neural net with 1 hidden layer
FANN::neural_net create_net(float learning_rate, unsigned int num_layers,unsigned int num_input, unsigned int num_hidden, unsigned int num_output){
FANN::neural_net net;
unsigned int layers[3]={num_input,num_hidden,num_output};
net.create_standard_array(num_layers,layers);
net.set_learning_rate(learning_rate);
net.set_activation_steepness_hidden(.1);
net.set_activation_steepness_output(.1);
net.set_activation_function_hidden(FANN::SIGMOID_SYMMETRIC_STEPWISE);
net.set_activation_function_output(FANN::SIGMOID_SYMMETRIC_STEPWISE);
//net.set_training_algorithm(FANN::TRAIN_INCREMENTAL);
// Set additional properties such as the training algorithm
//net.set_training_algorithm(FANN::TRAIN_QUICKPROP);
return net;
}
void train(Configuration *cfg)
{
QString fileName(QDir::homePath() + "/" + QCoreApplication::applicationName() + ".ini");
qDebug() << "using config file:" << fileName;
QSettings settings(fileName, QSettings::IniFormat);
const float learningRate = settings.value("learningRate", 0.8).toFloat();
const unsigned int numLayers = settings.value("numLayers", 3).toInt();
const unsigned int numInput = settings.value("numInput", 1024).toInt();
const unsigned int numHidden = settings.value("numHidden", 32).toInt();
const unsigned int numOutput = settings.value("numOutput", 1).toInt();
const float desiredError = settings.value("desiredError", 0.0001f).toFloat();
const unsigned int maxIterations = settings.value("maxIterations", 3000).toInt();
const unsigned int iterationsBetweenReports = settings.value("iterationsBetweenReports", 100).toInt();
FANN::neural_net net;
net.create_standard(numLayers, numInput, numHidden, numOutput);
net.set_learning_rate(learningRate);
net.set_activation_steepness_hidden(0.5);
net.set_activation_steepness_output(0.5);
net.set_learning_momentum(0.6);
net.set_activation_function_hidden(FANN::SIGMOID_SYMMETRIC);
net.set_activation_function_output(FANN::SIGMOID_SYMMETRIC);
net.set_training_algorithm(FANN::TRAIN_RPROP);
net.print_parameters();
FANN::training_data data;
if (data.read_train_from_file(cfg->getDataSavePath().toStdString()))
{
qDebug() << "Wczytano dane";
//inicjalizacja wag
net.init_weights(data);
data.shuffle_train_data();
net.set_callback(printCallback, NULL);
net.train_on_data(data, maxIterations,
iterationsBetweenReports, desiredError);
net.save(cfg->getNetSavePath().toStdString());
qDebug() << "Nauczono i zapisano siec";
}
}
int main(int argc, char *argv[])
{
if (argv[1][0] == 'r') {
WAVFile inp(argv[2]);
translate_wav(inp);
return 0;
}
if (argc == 1 || argc % 2 != 1) {
std::cout << "bad number of training examples\n";
return -1;
}
int to_open = (argc - 1)/2;
for (int i = 0; i < to_open; i++) {
WAVFile inp(argv[2*i+1]);
WAVFile out(argv[2*i+2]);
add_training_sound(inp, out);
}
float *train_in[input_training.size()];
float *train_out[output_training.size()];
for (int i = 0; i < input_training.size(); i++) {
train_in[i] = input_training[i];
train_out[i] = output_training[i];
}
FANN::training_data training;
training.set_train_data(input_training.size(), (samples_per_segment/2+1)*2,
train_in, (samples_per_segment/2+1)*2, train_out);
FANN::neural_net net;
const unsigned int layers[] = {(samples_per_segment/2+1)*2, (samples_per_segment/2+1)*2, (samples_per_segment/2+1)*2};
net.create_standard_array(3, (unsigned int*)layers);
net.set_activation_function_output(FANN::LINEAR);
//net.set_activation_function_hidden(FANN::LINEAR);
net.set_learning_rate(1.2f);
net.train_on_data(training, 50000, 1, 3.0f);
net.save("net.net");
}
void trainingThread::train()
{
std::stringstream log;
log << std::endl << " test started." << std::endl;
const float learning_rate = netConfigPTR->learning_rate ;
const unsigned int num_layers = netConfigPTR->num_layers;
const unsigned int num_input = netConfigPTR->num_input;
const unsigned int num_hidden = netConfigPTR->num_hidden;
const unsigned int num_output = netConfigPTR->num_output;
const float desired_error = netConfigPTR->desired_error;
const unsigned int max_iterations = netConfigPTR->max_iterations;
const unsigned int iterations_between_reports = netConfigPTR->iterations_between_reports;
log << std::endl << "Creating network." << std::endl;
FANN::neural_net net;
if (netConfigPTR->leyersVector.size() > 1)
{
unsigned int vectorSize = netConfigPTR->leyersVector.size();
unsigned int* leyers = new unsigned int[vectorSize+2];
leyers[0] = num_input;
for (unsigned int i = 0; i < vectorSize; ++i)
{
leyers[i+1] = netConfigPTR->leyersVector.at(i);
}
leyers[num_layers-1] = num_output;
for ( unsigned int i = 0 ; i< vectorSize+2 ; ++i)
{
qDebug() << "vector size: "<< vectorSize+2<<" i:"<<i<< " leyers "<< leyers[i];
}
net.create_standard_array(vectorSize+2, leyers);
//net.create_standard(vectorSize+2, leyers[0], leyers[2],leyers[3], leyers[1]);
delete[] leyers;
}
else
{
net.create_standard(num_layers, num_input, num_hidden, num_output);
}
net.set_learning_rate(learning_rate);
net.set_activation_steepness_hidden(1.0);
net.set_activation_steepness_output(1.0);
net.set_activation_function_hidden(FANN::SIGMOID_SYMMETRIC_STEPWISE);
net.set_activation_function_output(FANN::SIGMOID_SYMMETRIC_STEPWISE);
// Set additional properties such as the training algorithm
net.set_training_algorithm(netConfigPTR->trainingAlgo);
// Output network type and parameters
log << std::endl << "Network Type : ";
switch (net.get_network_type())
{
case FANN::LAYER:
log << "LAYER" << std::endl;
break;
case FANN::SHORTCUT:
log << "SHORTCUT" << std::endl;
break;
default:
log << "UNKNOWN" << std::endl;
break;
}
//net.print_parameters();
log << std::endl << "Training network." << std::endl;
FANN::training_data data;
if (data.read_train_from_file(netConfigPTR->trainingDataPatch))
{
// Initialize and train the network with the data
net.init_weights(data);
log << "Max Epochs " << std::setw(8) << max_iterations << ". "
<< "Desired Error: " << std::left << desired_error << std::right << std::endl;
emit updateLog(QString::fromStdString(log.str()));
log << "dupa";
log.str("");
log.clear();
net.set_callback(print_callback, nullptr);
net.train_on_data(data, max_iterations,
iterations_between_reports, desired_error);
log << std::endl << "Testing network." << std::endl;
for (unsigned int i = 0; i < data.length_train_data(); ++i)
{
// Run the network on the test data
fann_type *calc_out = net.run(data.get_input()[i]);
log << "test (";
for (unsigned int j = 0; j < num_input; ++j)
{
log << std::showpos << data.get_input()[i][j] << ", ";
//qDebug()<< "jestem w log<<";
}
log << ") -> " ;
for(unsigned int k = 0 ; k < num_output ; ++k)
{
log << calc_out[k] <<", ";
}
log << ",\t should be ";
for(unsigned int k = 0 ; k < num_output ; ++k)
{
log << data.get_output()[i][k] <<", ";
}
log << std::endl ;
}
log << std::endl << "Saving network." << std::endl;
// Save the network in floating point and fixed point
net.save(netConfigPTR->netFloat);
unsigned int decimal_point = net.save_to_fixed(netConfigPTR->netFixed);
std::string path = netConfigPTR->trainingDataPatch.substr(0,netConfigPTR->trainingDataPatch.size()-5);
data.save_train_to_fixed(path +"_fixed.data", decimal_point);
log << std::endl << "test completed." << std::endl;
emit updateLog(QString::fromStdString(log.str()));
emit updateProgressBar(100);
}
}
int callback_html(FANN::neural_net &rede_neural,
training_data &dados,
unsigned int turnos_max,
unsigned int turnos_por_amostra,
float margem_de_erro,
unsigned int turnos,
void *_cerebro){
static FILE *plot = NULL;
static float total = 0;
static float last_MSE = 0;
static float last_bit_fail = 0;
static int loops = 0;
static int MSE_counter = 0;
static int bit_fail_counter = 0;
static float steepness = 0;
static float aprendizagem = 0;
static float last_aprendizagem = 0;
static float aprendizagem_inicial = 0;
static bool set_steepness = false;
static bool set_aprendizagem = false;
static training_data *dados_teste = NULL;
char buffer[20000];
char MSE_buffer[64];
char bit_fail_buffer[64];
float diferenca_MSE;
float MSE;
int bit_fail = 0;
int diferenca_bit_fail = 0;
CCerebro *cerebro = (CCerebro*)_cerebro;
if(dados_teste){
//rede_neural.reset_MSE();
//printf("%f ", rede_neural.get_MSE());
MSE = rede_neural.test_data(*dados_teste);
//printf("%f\n", MSE);
}
MSE = rede_neural.get_MSE();
bit_fail = rede_neural.get_bit_fail();
//rede_neural.reset_MSE();
//printf("%f %f %d\n", MSE, last_MSE, (MSE - last_MSE) > 0.000001);
if(last_MSE)
diferenca_MSE = (last_MSE - MSE);
else
diferenca_MSE = 0;
if(last_bit_fail)
diferenca_bit_fail = last_bit_fail - bit_fail;
else
diferenca_bit_fail = 0;
if(last_bit_fail == bit_fail){
if(bit_fail_counter < BIT_FAIL_COUNTER)
bit_fail_counter++;
}else if(bit_fail_counter > 0)
bit_fail_counter--;// = bit_fail_counter * 0.85;
if(fabs(diferenca_MSE) < margem_de_erro){
if(MSE_counter < MSE_COUNTER)
MSE_counter++;
}else if(MSE_counter > 0)
MSE_counter--;// = MSE_counter * 0.9;
//printf("%d\n", bit_fail_counter);
//printf("%f %f\n",last_MSE, diferenca_MSE);
if(last_MSE){
fprintf(plot, "%d %f %f %d %d\n",turnos, MSE, diferenca_MSE, bit_fail_counter, MSE_counter);
fflush(plot);
total = total + diferenca_MSE;
}
last_MSE = MSE;
last_bit_fail = bit_fail;
if(set_aprendizagem){
if(!aprendizagem_inicial)
aprendizagem_inicial = cerebro->rede_neural->get_learning_rate();
aprendizagem = (1-((float)(MSE_counter + bit_fail_counter) / 200)) * aprendizagem_inicial;
if(!last_aprendizagem)
last_aprendizagem = aprendizagem;
if(last_aprendizagem > aprendizagem && !((MSE_counter + bit_fail_counter)%2)){
rede_neural.set_learning_rate(aprendizagem);
last_aprendizagem = aprendizagem;
}
}
if(set_steepness){
if(!steepness)
steepness = rede_neural.get_activation_steepness(1,0);
rede_neural.set_activation_steepness_layer(
(1-((float)(MSE_counter + bit_fail_counter) / 100))*steepness
,1);
//set_sneepness = true;
}
loops++;
if(turnos > 1){
if(loops < LOOPS && (bit_fail_counter < BIT_FAIL_COUNTER || MSE_counter < MSE_COUNTER) )
return 0;
}else{
bit_fail_counter = 0;
MSE_counter = 0;
last_aprendizagem = 0;
if(dados_teste)
free(dados_teste);
if(plot)
fclose(plot);
/*
dados_teste = new training_data();
if(cerebro->getDadosIdentificadosEscalados()){
dados_teste->set_train_data(cerebro->num_dados_identificados,
cerebro->num_caracteristicas, cerebro->dados_identificados_escalados,
cerebro->num_caracteristicas, cerebro->dados_identificados_escalados);
printf("Utilizando massa de testes identificada com %d casos.\n",cerebro->num_dados_identificados );
}else if(cerebro->getDadosEscalados()){
dados_teste->set_train_data(cerebro->num_dados,
cerebro->num_caracteristicas, cerebro->dados_escalados,
cerebro->num_caracteristicas, cerebro->dados_escalados);
printf("Utilizando massa de testes não identificada com %d casos.\n",cerebro->num_dados );
}else{
printf("Sem dados para treinar.\n");
delete dados_teste;
return -1;
}
*/
printf("Plotando em %s.\n",cerebro->plot_file);
plot = fopen(cerebro->plot_file, "w");
}
loops = 0;
memset(buffer,0,20000);
//clear_screen();
cor_nivel(bit_fail_counter,bit_fail_buffer);
cor_nivel(MSE_counter,MSE_buffer);
//printf("%f\n",round((float)bit_fail_counter/10));
sprintf((char*)(buffer + strlen(buffer)), "%08d turnos, precisão %1.8f(%-1.8f)(%s%%), nivel de adaptacao %d(%d)(%s%%)",
turnos, MSE, diferenca_MSE, MSE_buffer, bit_fail, diferenca_bit_fail, bit_fail_buffer);
sprintf((char*)(buffer + strlen(buffer)),", depuracao %-1.6f, aprendizagem %f, passo %f", total,last_aprendizagem,steepness);
sprintf((char*)(buffer + strlen(buffer)),"\n");
printf(buffer);
//textcolor(RESET, WHITE, BLACK);
//rede_neural.print_connections(buffer);
//printf(buffer);
if(bit_fail_counter == BIT_FAIL_COUNTER && MSE_counter == MSE_COUNTER){
printf("Treino completado em %08d, adaptacao %d, MSE %f\n",turnos, bit_fail, MSE);
return -1;
}
return 0;
}