Function benchmark_nb

examples/machine_learning/naive_bayes.cpp:80–102

}

void benchmark_nb(const array &train_feats, const array test_feats,
                  const array &train_labels, int num_classes) {
    array mu, sig2;
    int iter      = 25;
    float *priors = new float[num_classes];

    timer::start();
    for (int i = 0; i < iter; i++) {
        naive_bayes_train(priors, mu, sig2, train_feats, train_labels,
                          num_classes);
    }
    af::sync();
    printf("Training time: %4.4lf s\n", timer::stop() / iter);

    timer::start();
    for (int i = 0; i < iter; i++) {
        naive_bayes_predict(priors, mu, sig2, test_feats, num_classes);
    }
    af::sync();
    printf("Prediction time: %4.4lf s\n", timer::stop() / iter);

    delete[] priors;
}

void naive_bayes_demo(bool console, int perc) {
    array train_images, train_labels;