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

Corrected a bug in the one hot representation

parent 07d37a0d
No related branches found
No related tags found
No related merge requests found
...@@ -105,8 +105,8 @@ std::vector<Layer> create_layers_lemmatizer(unsigned int nb_inputs, unsigned int ...@@ -105,8 +105,8 @@ std::vector<Layer> create_layers_lemmatizer(unsigned int nb_inputs, unsigned int
std::vector<Layer> create_layers_parser(unsigned int nb_inputs, unsigned int nb_classes) std::vector<Layer> create_layers_parser(unsigned int nb_inputs, unsigned int nb_classes)
{ {
std::vector<Layer> layers{ std::vector<Layer> layers{
{nb_inputs, 500, 0.2, RELU}, {nb_inputs, 100, 0.2, RELU},
{500, nb_classes, 0.0, LINEAR} {100, nb_classes, 0.0, LINEAR}
}; };
return layers; return layers;
......
...@@ -187,6 +187,7 @@ void feat_vec_fill_input_array_dnn(std::vector<float> & input, ...@@ -187,6 +187,7 @@ void feat_vec_fill_input_array_dnn(std::vector<float> & input,
input[index + value] = 1.0; input[index + value] = 1.0;
} }
index += feature_size;
continue; continue;
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment