rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add AUC computing in Experiment and store in result

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-07-12 17:23:03 +02:00
parent 26dfe6d056
commit 84adf13a79
7 changed files with 134 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/Scores.cpp
View File

@@ -2,12 +2,13 @@
#include "Scores.h"
#include "common/Colors.h"
namespace platform {
Scores::Scores(torch::Tensor& y_test, torch::Tensor& y_pred, int num_classes, std::vector<std::string> labels) : num_classes(num_classes), labels(labels)
Scores::Scores(torch::Tensor& y_test, torch::Tensor& y_proba, int num_classes, std::vector<std::string> labels) : num_classes(num_classes), labels(labels)
{
if (labels.size() == 0) {
init_default_labels();
}
total = y_test.size(0);
auto y_pred = y_proba.argmax(1);
accuracy_value = (y_pred == y_test).sum().item<float>() / total;
init_confusion_matrix();
for (int i = 0; i < total; i++) {