create an evaluation function for macro f1 score in matlab using the confusion matrix in matlab

main.m
% Assuming you have a confusion matrix named "confusionMatrix"
numClasses = size(confusionMatrix, 1);
precision = zeros(numClasses, 1);
recall = zeros(numClasses, 1);
for i = 1:numClasses
    tp = confusionMatrix(i, i);
    fp = sum(confusionMatrix(:, i)) - tp;
    fn = sum(confusionMatrix(i, :)) - tp;
    precision(i) = tp / (tp + fp);
    recall(i) = tp / (tp + fn);
end
375 chars
12 lines

main.m
f1Score = 2 * (precision .* recall) ./ (precision + recall);
61 chars
2 lines

main.m
macroF1 = mean(f1Score);
25 chars
2 lines

main.m
function macroF1 = calculateMacroF1(confusionMatrix)
    numClasses = size(confusionMatrix, 1);
    precision = zeros(numClasses, 1);
    recall = zeros(numClasses, 1);
    for i = 1:numClasses
        tp = confusionMatrix(i, i);
        fp = sum(confusionMatrix(:, i)) - tp;
        fn = sum(confusionMatrix(i, :)) - tp;
        precision(i) = tp / (tp + fp);
        recall(i) = tp / (tp + fn);
    end
    f1Score = 2 * (precision .* recall) ./ (precision + recall);
    macroF1 = mean(f1Score);
end
503 chars
15 lines

main.m
confusionMatrix = ...; % your confusion matrix here
macroF1 = calculateMacroF1(confusionMatrix);
fprintf('Macro F1 Score: %.2f\n', macroF1);
141 chars
4 lines