doc_classifier.h

使用具有增量学习的监控式学习方法。包括几个不同的分类算法。

#ifndef DOC_CLASSIFIER_H
#define DOC_CLASSIFIER_H

#include <stdio.h>
#include "document.h"
#include "vectorizer.h"

/**
 * Classifier type
 */
enum cl_type {
    DOCUMENT,		/**< Document classifier */
    VECTOR		/**< Vector classifier */
};

/**
 * Document classifier functions.
 */
typedef struct {
    void *(*new) (const char *);
    void *(*load) (FILE *);
    int (*save) (FILE *, void *);
    void (*free) (void *);
    //int (*set_options) (void *, char *);
    int (*info) (void *);	/**< Currently: Get number of classes */
    int (*learn) (void *, void *, int);
    int (*unlearn) (void *, void *, int);
    int (*remove) (void *, void *);
    int (*classify_top) (void *, void *);
    int *(*classify_rank) (void *, void *);
    double *(*classify_score) (void *, void *);
} doc_classifier_functions;

typedef struct doc_classifier_ doc_classifier;

doc_classifier *
doc_classifier_new (void *data, const doc_classifier_functions *funcs,
		    vectorizer *vec, enum cl_type clt);

vectorizer *
doc_classifier_get_vectorizer (doc_classifier *cl);

int
doc_classifier_get_noc (doc_classifier *cl);

int
doc_classifier_learn (doc_classifier *cl, document *d, int class);

int
doc_classifier_unlearn (doc_classifier *cl, document *d, int class);

int
doc_classifier_classify_top (doc_classifier *cl, document *d);

double *
doc_classifier_classify_score (doc_classifier *cl, document *d);

int *
doc_classifier_classify_rank (doc_classifier *cl, document *d);

int
doc_classifier_save (FILE *fp, doc_classifier *cl);

#endif