-
Notifications
You must be signed in to change notification settings - Fork 0
/
training.h
30 lines (24 loc) · 1.19 KB
/
training.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#ifndef TRAINING
#define TRAINING
#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <map>
#include <cmath>
#include <random>
#include <algorithm>
#include <Eigen/Dense>
#include "classic.h"
#include "LMs.h"
#include "perso.h"
#include "incremental.h"
#include "essai.h"
std::map<std::string,double> train(Eigen::MatrixXd& X, Eigen::MatrixXd& Y, int const& L, std::vector<int> const& nbNeurons, std::vector<int> const& globalIndices,
std::vector<std::string> const& activations, std::vector<Eigen::MatrixXd>& weights, std::vector<Eigen::VectorXd>& bias, std::string const& type_perte,
std::string const& famille_algo, std::string const& algo,double const& eps, int const& maxEpoch, double const& learning_rate,
double const& clip,double const& seuil, double const& beta1, double const& beta2,int const& batch_size,
double& mu, double& factor, double const& RMin, double const& RMax, int const& b, double const& alpha,
double const& pas, double const& Rlim, double& factorMin, double const& power, double const& alphaChap, double const& epsDiag,
bool const tracking=false, bool const track_continuous=false, bool const record=false, std::string const fileExtension="");
#endif