gensvm_predict.c

Go to the documentation of this file.

#include "gensvm_predict.h"

void gensvm_predict_labels(struct GenData *testdata, struct GenModel *model,
        long *predy)
{
    long i, j, k, n, K, label;
    double norm, min_dist,
           *S = NULL,
           *ZV = NULL;

    n = testdata->n;
    K = model->K;

    // allocate necessary memory
    S = Calloc(double, K-1);
    ZV = Calloc(double, n*(K-1));

    // Generate the simplex matrix
    gensvm_simplex(model);

    // Generate the simplex space vectors
    gensvm_calculate_ZV(model, testdata, ZV);

    // Calculate the distance to each of the vertices of the simplex.
    // The closest vertex defines the class label
    for (i=0; i<n; i++) {
        label = 0;
        min_dist = INFINITY;
        for (j=0; j<K; j++) {
            for (k=0; k<K-1; k++) {
                S[k] = matrix_get(ZV, K-1, i, k) -
                    matrix_get(model->U, K-1, j, k);
            }
            norm = cblas_dnrm2(K-1, S, 1);
            if (norm < min_dist) {
                label = j+1;
                min_dist = norm;
            }
        }
        predy[i] = label;
    }

    free(ZV);
    free(S);
}

double gensvm_prediction_perf(struct GenData *data, long *predy)
{
    long i, correct = 0;
    double performance;

    for (i=0; i<data->n; i++)
        if (data->y[i] == predy[i])
            correct++;

    performance = ((double) correct)/((double) data->n)* 100.0;

    return performance;
}