Skip to content
Snippets Groups Projects
Commit 3cf39b6d authored by Cédric Traizet's avatar Cédric Traizet
Browse files

ENH: prediction field is now an attribute

parent e2ad78e8
Branches
Tags
No related merge requests found
...@@ -170,7 +170,7 @@ protected: ...@@ -170,7 +170,7 @@ protected:
//Second Extraction //Second Extraction
sampleExtraction->SetParameterInputImage("in", predictorImageList->GetNthElement(i)); sampleExtraction->SetParameterInputImage("in", predictorImageList->GetNthElement(i));
sampleExtraction->SetParameterString("outfield", "list"); sampleExtraction->SetParameterString("outfield", "list");
sampleExtraction->SetParameterStringList("outfield.list.names", {"prediction"}); sampleExtraction->SetParameterStringList("outfield.list.names", {m_PredictionFieldName});
ExecuteInternal("extraction"); ExecuteInternal("extraction");
} }
} }
...@@ -301,6 +301,9 @@ private: ...@@ -301,6 +301,9 @@ private:
} }
std::string m_ClassFieldName = "regclass"; std::string m_ClassFieldName = "regclass";
std::string m_PredictionFieldName = "prediction";
std::map< std::string, std::vector<std::string>> m_FileHandler; std::map< std::string, std::vector<std::string>> m_FileHandler;
}; };
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment