-
Notifications
You must be signed in to change notification settings - Fork 1
/
setlabel.c
161 lines (134 loc) · 5.06 KB
/
setlabel.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
/************************************************************************
* *
* Program package 'lvq_pak': *
* *
* setlabel.c *
* -sets the labels of entries by the majority voting *
* *
* Version 3.0 *
* Date: 1 Mar 1995 *
* *
* NOTE: This program package is copyrighted in the sense that it *
* may be used for scientific purposes. The package as a whole, or *
* parts thereof, cannot be included or used in any commercial *
* application without written permission granted by its producents. *
* No programs contained in this package may be copied for commercial *
* distribution. *
* *
* All comments concerning this program package may be sent to the *
* e-mail address 'lvq@cochlea.hut.fi'. *
* *
************************************************************************/
#include <stdio.h>
#include <float.h>
#include "lvq_pak.h"
#include "datafile.h"
static char *usage[] = {
"setlabel - sets the labels of codebook entries by the majority voting\n",
"Required parameters:\n",
" -cin filename input codebook file\n",
" -din filename input data\n",
" -cout filename output codebook filename\n",
"Optional parameters:\n",
" -knn N use N nearest neighbours (default: 5)\n",
" -buffer integer buffered reading of data, integer lines at a time\n",
" -selfuncs name select a set of functions\n",
NULL};
struct entries *find_labels(struct teach_params *teach)
{
long noc, i;
struct data_entry *codetmp;
WINNER_FUNCTION *find_winners = teach->winner;
struct entries *data = teach->data;
struct entries *codes = teach->codes;
int knn = teach->knn;
struct winner_info *winners;
struct hitlist *hits;
eptr p;
if ((hits = new_hitlist()) == NULL)
{
fprintf(stderr, "can't create hitlist\n");
return NULL;
}
winners = calloc(knn, sizeof(struct winner_info));
if (knn < 1)
knn = 1;
/* Scan all codebook entries */
codetmp = rewind_entries(codes, &p);
noc = codes->num_entries;
while (codetmp != NULL) {
clear_hitlist(hits);
/* Find best matching data vectors */
find_winners(data, codetmp, winners, knn);
/* Find the majority of votes */
for (i = 0; i < knn; i++)
add_hit(hits, get_entry_label(winners[i].winner));
set_entry_label(codetmp, hits->head->label);
/* Take the next code entry */
codetmp = next_entry(&p);
ifverbose(1)
mprint((long) noc--);
}
ifverbose(1)
{
mprint((long) 0);
fprintf(stderr, "\n");
}
free_hitlist(hits);
return(codes);
}
int main(int argc, char **argv)
{
int knn;
char *in_data_file;
char *in_code_file;
char *out_code_file;
struct entries *data, *codes;
struct teach_params params;
long buffer;
char *funcname = NULL;
global_options(argc, argv);
if (extract_parameter(argc, argv, "-help", OPTION2))
{
printhelp();
exit(0);
}
in_data_file = extract_parameter(argc, argv, IN_DATA_FILE, ALWAYS);
in_code_file = extract_parameter(argc, argv, IN_CODE_FILE, ALWAYS);
out_code_file = extract_parameter(argc, argv, OUT_CODE_FILE, ALWAYS);
knn = oatoi(extract_parameter(argc, argv, KNN_NEIGHBORS, OPTION), 5);
buffer = oatoi(extract_parameter(argc, argv, "-buffer", OPTION), 0);
funcname = extract_parameter(argc, argv, "-selfuncs", OPTION);
ifverbose(2)
fprintf(stderr, "Input entries are read from file %s\n", in_data_file);
if ((data = open_entries(in_data_file)) == NULL)
{
fprintf(stderr, "Can't open data file '%s'\n", in_data_file);
exit(1);
}
label_not_needed(1);
ifverbose(2)
fprintf(stderr, "Codebook entries are read from file %s\n", in_code_file);
if ((codes = open_entries(in_code_file)) == NULL)
{
fprintf(stderr, "Can't open codes file '%s'\n", in_code_file);
close_entries(data);
exit(1);
}
if (data->dimension != codes->dimension) {
fprintf(stderr, "Data and codebook vectors have different dimensions");
close_entries(data);
close_entries(codes);
exit(1);
}
set_teach_params(¶ms, codes, data, 0, funcname);
params.knn = knn;
params.winner = find_winner_knn;
codes = find_labels(¶ms);
ifverbose(2)
fprintf(stderr, "Codebook entries are saved to file %s\n", out_code_file);
save_entries(codes, out_code_file);
close_entries(data);
close_entries(codes);
return(0);
}