Added function so set columns of NeuralNetwork

torch_modules/include/NeuralNetwork.hpp

@@ -12,7 +12,7 @@ class NeuralNetworkImpl : public torch::nn::Module
   int leftBorder{5};
   int rightBorder{5};
   int nbStackElements{2};
-  std::vector<std::string> columns{"FORM", "UPOS"};
+  std::vector<std::string> columns{"FORM"};
 
   protected :
 
@@ -25,6 +25,7 @@ class NeuralNetworkImpl : public torch::nn::Module
   virtual torch::Tensor forward(torch::Tensor input) = 0;
   std::vector<long> extractContext(Config & config, Dict & dict) const;
   int getContextSize() const;
+  void setColumns(const std::vector<std::string> & columns);
 };
 TORCH_MODULE(NeuralNetwork);
 

torch_modules/src/ConcatWordsNetwork.cpp

@@ -6,6 +6,7 @@ ConcatWordsNetworkImpl::ConcatWordsNetworkImpl(int nbOutputs, int leftBorder, in
   setLeftBorder(leftBorder);
   setRightBorder(rightBorder);
   setNbStackElements(nbStackElements);
+  setColumns({"FORM", "UPOS"});
 
   wordEmbeddings = register_module("word_embeddings", torch::nn::Embedding(torch::nn::EmbeddingOptions(50000, embeddingsSize)));
   linear1 = register_module("linear1", torch::nn::Linear(getContextSize()*embeddingsSize, 500));

torch_modules/src/NeuralNetwork.cpp

@@ -56,3 +56,8 @@ void NeuralNetworkImpl::setNbStackElements(int nbStackElements)
   this->nbStackElements = nbStackElements;
 }
 
+void NeuralNetworkImpl::setColumns(const std::vector<std::string> & columns)
+{
+  this->columns = columns;
+}
+

torch_modules/src/OneWordNetwork.cpp

@@ -19,6 +19,7 @@ OneWordNetworkImpl::OneWordNetworkImpl(int nbOutputs, int focusedIndex)
   setLeftBorder(leftBorder);
   setRightBorder(rightBorder);
   setNbStackElements(0);
+  setColumns({"FORM", "UPOS"});
 }
 
 torch::Tensor OneWordNetworkImpl::forward(torch::Tensor input)