Refactor aggregate score to a constructor
This commit is contained in:
@@ -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