@@ -410,10 +410,7 @@ Index NeuralNetwork::get_inputs_number() const
410410
411411
412412 const Shape input_shape = layers[0 ]->get_input_shape ();
413- cout << " input_shape rank: " << input_shape.rank << endl;
414- for (size_t i = 0 ; i < input_shape.rank ; i++)
415- cout << " input_shape[" << i << " ] = " << input_shape.shape [i] << endl;
416- cout << " count: " << input_shape.count () << endl;
413+
417414 return input_shape.count ();
418415}
419416
@@ -971,19 +968,12 @@ Tensor<string, 2> NeuralNetwork::get_dense2d_layers_information() const
971968}
972969
973970
974-
975971void NeuralNetwork::to_XML (XMLPrinter& printer) const
976972{
977- cout << " [to_XML] ENTER" << endl;
978-
979973 const Index inputs_number = get_inputs_number ();
980974 const Index layers_number = get_layers_number ();
981975 const Index outputs_number = get_outputs_number ();
982976
983- cout << " [to_XML] inputs=" << inputs_number
984- << " layers=" << layers_number
985- << " outputs=" << outputs_number << endl;
986-
987977 vector<string> input_names = get_input_feature_names ();
988978 while (input_names.size () < static_cast <size_t >(inputs_number))
989979 input_names.push_back (" input_" + to_string (input_names.size () + 1 ));
@@ -992,9 +982,6 @@ void NeuralNetwork::to_XML(XMLPrinter& printer) const
992982 while (output_names.size () < static_cast <size_t >(outputs_number))
993983 output_names.push_back (" output_" + to_string (output_names.size () + 1 ));
994984
995- cout << " [to_XML] input_names.size=" << input_names.size ()
996- << " output_names.size=" << output_names.size () << endl;
997-
998985 printer.OpenElement (" NeuralNetwork" );
999986
1000987 // Inputs
@@ -1003,64 +990,43 @@ void NeuralNetwork::to_XML(XMLPrinter& printer) const
1003990 for (Index i = 0 ; i < inputs_number; i++)
1004991 add_xml_element_attribute (printer, " Input" , input_names[i], " Index" , to_string (i + 1 ));
1005992 printer.CloseElement ();
1006- cout << " [to_XML] Inputs section done" << endl;
1007993
1008994 // Layers
1009995 printer.OpenElement (" Layers" );
1010996 add_xml_element (printer, " LayersNumber" , to_string (layers_number));
1011997
1012- cout << " [to_XML] Starting layers loop, layers_number=" << layers_number << endl;
1013998 for (Index i = 0 ; i < layers_number; i++)
1014- {
1015- cout << " [to_XML] Layer " << i
1016- << " name=" << layers[i]->get_name ();
1017999 layers[i]->to_XML (printer);
1018- cout << " [to_XML] Layer " << i << " serialized OK" << endl;
1019- }
1020- cout << " [to_XML] Layers loop done" << endl;
1021-
1022- // LayerInputIndices
1023- cout << " [to_XML] layer_input_indices.size=" << layer_input_indices.size ()
1024- << " layers_number=" << layers_number << endl;
1025-
1026- if (Index (layer_input_indices.size ()) != layers_number)
1027- cout << " [to_XML] *** WARNING MISMATCH layer_input_indices vs layers_number ***" << endl;
10281000
10291001 printer.OpenElement (" LayerInputIndices" );
10301002 for (Index i = 0 ; i < Index (layer_input_indices.size ()); i++)
10311003 {
1032- cout << " [to_XML] indices[" << i << " ]="
1033- << vector_to_string (layer_input_indices[i]) << endl;
10341004 add_xml_element_attribute (printer, " LayerInputsIndices" ,
10351005 vector_to_string (layer_input_indices[i]), " LayerIndex" , to_string (i));
10361006 }
10371007 printer.CloseElement ();
10381008 printer.CloseElement (); // Layers
1039- cout << " [to_XML] LayerInputIndices done" << endl;
10401009
10411010 // Outputs
10421011 printer.OpenElement (" Outputs" );
10431012 const Index outputs_count = has (" Embedding" ) ? outputs_number : output_names.size ();
1044- cout << " [to_XML] outputs_count=" << outputs_count << endl;
10451013 add_xml_element (printer, " OutputsNumber" , to_string (outputs_count));
10461014 for (Index i = 0 ; i < outputs_count; i++)
10471015 add_xml_element_attribute (printer, " Output" , output_names[i], " Index" , to_string (i + 1 ));
10481016 printer.CloseElement ();
1049- cout << " [to_XML] Outputs done" << endl;
10501017
10511018 add_xml_element (printer, " Display" , to_string (display));
10521019
10531020 // Parameters
10541021 printer.OpenElement (" Parameters" );
1055- cout << " [to_XML] parameters_number=" << get_parameters_number () << endl;
10561022 if (get_parameters_number () > 0 )
10571023 printer.PushText (vector_to_string (parameters, " " ).c_str ());
10581024 printer.CloseElement ();
10591025
10601026 printer.CloseElement (); // NeuralNetwork
1061- cout << " [to_XML] EXIT OK" << endl;
10621027}
10631028
1029+
10641030void NeuralNetwork::from_XML (const XMLDocument& document)
10651031{
10661032 const XMLElement* neural_network_element = document.FirstChildElement (" NeuralNetwork" );
0 commit comments