Skip to content
Snippets Groups Projects
Commit ecc8a003 authored by Ludovic Hussonnois's avatar Ludovic Hussonnois
Browse files

ENH: Compute and write contingency table.

parent 9417aa1a
Branches
Tags
No related merge requests found
......@@ -18,6 +18,7 @@
// Validation
#include "otbConfusionMatrixCalculator.h"
#include "otbContingencyTableCalculator.h"
namespace otb
{
......@@ -81,11 +82,27 @@ private:
}
else
{
// TODO Compute Contingency Table
WriteContingencyTable();
}
}
void WriteContingencyTable()
{
// Compute contingency table
typedef ContingencyTableCalculator<ClassLabelType> ContigencyTableCalcutaltorType;
ContigencyTableCalcutaltorType::Pointer contingencyTableCalculator = ContigencyTableCalcutaltorType::New();
contingencyTableCalculator->Compute(predictedList->Begin(), predictedList->End(),
classificationListSamples.labeledListSample->Begin(),
classificationListSamples.labeledListSample->End());
ContingencyTable<ClassLabelType> contingencyTable = contingencyTableCalculator->GetContingencyTable();
// Write contingency table
std::ofstream outFile;
outFile.open( this->GetParameterString( "io.confmatout" ).c_str() );
outFile << contingencyTable.to_csv();
}
ConfusionMatrixCalculatorType::Pointer
ComputeConfusionMatrix(const TargetListSampleType::Pointer &predictedListSample,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment