Joseph Redmon
2015-10-09 c40cdeb4021fc1a638969563972f13c9f5e90d74
src/network.c
@@ -540,12 +540,12 @@
    return acc;
}
float *network_accuracies(network net, data d)
float *network_accuracies(network net, data d, int n)
{
    static float acc[2];
    matrix guess = network_predict_data(net, d);
    acc[0] = matrix_topk_accuracy(d.y, guess,1);
    acc[1] = matrix_topk_accuracy(d.y, guess,5);
    acc[1] = matrix_topk_accuracy(d.y, guess, n);
    free_matrix(guess);
    return acc;
}