diff --git a/openbr/core/classify.cpp b/openbr/core/classify.cpp index 61e0cd2..200bc69 100644 --- a/openbr/core/classify.cpp +++ b/openbr/core/classify.cpp @@ -66,11 +66,12 @@ void br::EvalClassification(const QString &predictedInput, const QString &truthI counters[subject].falsePositive += 1.f / predictedSubjects.size(); } - QSharedPointer output(Output::make("", FileList() << "Subject" << "Count" << "Precision" << "Recall" << "F-score", FileList(counters.size()))); + const QStringList keys = counters.keys(); + QSharedPointer output(Output::make("", FileList() << "Count" << "Precision" << "Recall" << "F-score", FileList(keys))); int tpc = 0; int fnc = 0; - const QStringList keys = counters.keys(); + for (int i=0; isetRelative(File("", subject).get("Label"), i, 0); - output->setRelative(count, i, 1); - output->setRelative(precision, i, 2); - output->setRelative(recall, i, 3); - output->setRelative(fscore, i, 4); + output->setRelative(count, i, 0); + output->setRelative(precision, i, 1); + output->setRelative(recall, i, 2); + output->setRelative(fscore, i, 3); } qDebug("Overall Accuracy = %f", (float)tpc / (float)(tpc + fnc));