Skip to content
Snippets Groups Projects
Commit 1a6c5895 authored by Franck Dary's avatar Franck Dary
Browse files

fine tuned mlp parameters for a good baseline

parent 7907a9dd
No related branches found
No related tags found
No related merge requests found
...@@ -75,8 +75,8 @@ void train_perceptron(context * ctx){ ...@@ -75,8 +75,8 @@ void train_perceptron(context * ctx){
std::vector<Layer> create_layers_tagger(unsigned int nb_inputs, unsigned int nb_classes) std::vector<Layer> create_layers_tagger(unsigned int nb_inputs, unsigned int nb_classes)
{ {
std::vector<Layer> layers{ std::vector<Layer> layers{
{nb_inputs, 100, 0.2, RELU}, {nb_inputs, 200, 0.3, RELU},
{100, nb_classes, 0.0, LINEAR} {200, nb_classes, 0.0, LINEAR}
}; };
return layers; return layers;
...@@ -85,8 +85,8 @@ std::vector<Layer> create_layers_tagger(unsigned int nb_inputs, unsigned int nb_ ...@@ -85,8 +85,8 @@ std::vector<Layer> create_layers_tagger(unsigned int nb_inputs, unsigned int nb_
std::vector<Layer> create_layers_morpho(unsigned int nb_inputs, unsigned int nb_classes) std::vector<Layer> create_layers_morpho(unsigned int nb_inputs, unsigned int nb_classes)
{ {
std::vector<Layer> layers{ std::vector<Layer> layers{
{nb_inputs, 200, 0.2, RELU}, {nb_inputs, 350, 0.3, RELU},
{200, nb_classes, 0.0, LINEAR} {350, nb_classes, 0.0, LINEAR}
}; };
return layers; return layers;
...@@ -95,8 +95,8 @@ std::vector<Layer> create_layers_morpho(unsigned int nb_inputs, unsigned int nb_ ...@@ -95,8 +95,8 @@ std::vector<Layer> create_layers_morpho(unsigned int nb_inputs, unsigned int nb_
std::vector<Layer> create_layers_lemmatizer(unsigned int nb_inputs, unsigned int nb_classes) std::vector<Layer> create_layers_lemmatizer(unsigned int nb_inputs, unsigned int nb_classes)
{ {
std::vector<Layer> layers{ std::vector<Layer> layers{
{nb_inputs, 300, 0.2, RELU}, {nb_inputs, 250, 0.2, RELU},
{300, nb_classes, 0.0, LINEAR} {250, nb_classes, 0.0, LINEAR}
}; };
return layers; return layers;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment