diff options
Diffstat (limited to 'src/classifiers/classifiers.h')
| -rw-r--r-- | src/classifiers/classifiers.h | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/classifiers/classifiers.h b/src/classifiers/classifiers.h index 12787f049..de937bc3f 100644 --- a/src/classifiers/classifiers.h +++ b/src/classifiers/classifiers.h @@ -14,13 +14,20 @@ struct classifier_ctx { GHashTable *results; struct classifier_config *cfg; }; + +struct classify_weight { + const char *name; + double weight; +}; + /* Common classifier structure */ struct classifier { char *name; struct classifier_ctx* (*init_func)(memory_pool_t *pool, struct classifier_config *cf); void (*classify_func)(struct classifier_ctx* ctx, statfile_pool_t *pool, GTree *input, struct worker_task *task); void (*learn_func)(struct classifier_ctx* ctx, statfile_pool_t *pool, - stat_file_t *file, GTree *input, gboolean in_class, double *sum); + stat_file_t *file, GTree *input, gboolean in_class, double *sum, double multiplier); + GList* (*weights_func)(struct classifier_ctx* ctx, statfile_pool_t *pool, GTree *input, struct worker_task *task); }; /* Get classifier structure by name or return NULL if this name is not found */ @@ -29,7 +36,10 @@ struct classifier* get_classifier (char *name); /* Winnow algorithm */ struct classifier_ctx* winnow_init (memory_pool_t *pool, struct classifier_config *cf); void winnow_classify (struct classifier_ctx* ctx, statfile_pool_t *pool, GTree *input, struct worker_task *task); -void winnow_learn (struct classifier_ctx* ctx, statfile_pool_t *pool, stat_file_t *file, GTree *input, gboolean in_class, double *sum); +void winnow_learn (struct classifier_ctx* ctx, statfile_pool_t *pool, stat_file_t *file, GTree *input, + gboolean in_class, double *sum, double multiplier); +GList *winnow_weights (struct classifier_ctx* ctx, statfile_pool_t *pool, GTree *input, struct worker_task *task); + /* Array of all defined classifiers */ extern struct classifier classifiers[]; |