Skip to content
Snippets Groups Projects
Commit aaed0671 authored by Julien Michel's avatar Julien Michel
Browse files

ENH: Split WriteContingencyTable into ComputeContingencyTable and WriteContingencyTable

parent 0d10761f
No related branches found
No related tags found
No related merge requests found
...@@ -83,25 +83,34 @@ private: ...@@ -83,25 +83,34 @@ private:
} }
else else
{ {
WriteContingencyTable(); ContingencyTable<ClassLabelType> table = ComputeContingencyTable(predictedList,classificationListSamples.labeledListSample);
WriteContingencyTable(table);
} }
} }
ContingencyTable<ClassLabelType> ComputeContingencyTable(const TargetListSampleType::Pointer &predictedListSample,
void WriteContingencyTable() const TargetListSampleType::Pointer &performanceLabeledListSample)
{ {
// Compute contingency table
typedef ContingencyTableCalculator<ClassLabelType> ContigencyTableCalcutaltorType; typedef ContingencyTableCalculator<ClassLabelType> ContigencyTableCalcutaltorType;
ContigencyTableCalcutaltorType::Pointer contingencyTableCalculator = ContigencyTableCalcutaltorType::New(); ContigencyTableCalcutaltorType::Pointer contingencyTableCalculator = ContigencyTableCalcutaltorType::New();
contingencyTableCalculator->Compute(predictedList->Begin(), predictedList->End(), contingencyTableCalculator->Compute(performanceLabeledListSample->Begin(),
classificationListSamples.labeledListSample->Begin(), performanceLabeledListSample->End(),predictedListSample->Begin(), predictedListSample->End());
classificationListSamples.labeledListSample->End());
ContingencyTable<ClassLabelType> contingencyTable = contingencyTableCalculator->GetContingencyTable(); return contingencyTableCalculator->GetContingencyTable();
}
// Write contingency table
std::ofstream outFile;
outFile.open( this->GetParameterString( "io.confmatout" ).c_str() ); void WriteContingencyTable(const ContingencyTable<ClassLabelType> & table)
outFile << contingencyTable.to_csv(); {
if(IsParameterEnabled("io.confmatout"))
{
// Write contingency table
std::ofstream outFile;
outFile.open( this->GetParameterString( "io.confmatout" ).c_str() );
outFile << table.to_csv();
}
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment