Refactor aggregate score to a constructor
This commit is contained in:
@@ -40,6 +40,15 @@ namespace platform {
|
||||
}
|
||||
compute_accuracy_value();
|
||||
}
|
||||
static Score 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]);
|
||||
scores.aggregate(score);
|
||||
}
|
||||
return scores;
|
||||
}
|
||||
void Scores::compute_accuracy_value()
|
||||
{
|
||||
accuracy_value = 0;
|
||||
|
@@ -10,6 +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);
|
||||
float accuracy();
|
||||
float f1_score(int num_class);
|
||||
float f1_weighted();
|
||||
|
Reference in New Issue
Block a user