Fix refactor of static aggregate method

This commit is contained in:
2024-05-17 23:38:21 +02:00
parent 49a36904dc
commit c165a4bdda
3 changed files with 5 additions and 6 deletions

View File

@@ -40,11 +40,11 @@ namespace platform {
}
compute_accuracy_value();
}
static Score Scores::create_aggregate(json& data, std::string key)
Scores Scores::create_aggregate(json& data, std::string key)
{
auto scores = Scores(result[key][0]);
for (int i = 1; i < result[key].size(); i++) {
auto score = Scores(result[key][i]);
auto scores = Scores(data[key][0]);
for (int i = 1; i < data[key].size(); i++) {
auto score = Scores(data[key][i]);
scores.aggregate(score);
}
return scores;

View File

@@ -10,7 +10,7 @@ namespace platform {
public:
Scores(torch::Tensor& y_test, torch::Tensor& y_pred, int num_classes, std::vector<std::string> labels = {});
explicit Scores(json& confusion_matrix_);
static Score create_aggregate(json& data, std::string key);
static Scores create_aggregate(json& data, std::string key);
float accuracy();
float f1_score(int num_class);
float f1_weighted();

View File

@@ -6,7 +6,6 @@
#include "ReportBase.h"
#include "main/Scores.h"
namespace platform {
const int MAXL = 133;
class ReportConsole : public ReportBase {