blob: a26a7505c18c10babd36736cac6bfa954ebae3b7 (
plain)
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
|
package de.lmu.ifi.dbs.elki.algorithm.outlier.distance.parallel;
/*
This file is part of ELKI:
Environment for Developing KDD-Applications Supported by Index-Structures
Copyright (C) 2014
Ludwig-Maximilians-Universität München
Lehr- und Forschungseinheit für Datenbanksysteme
ELKI Development Team
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
import de.lmu.ifi.dbs.elki.database.ids.DBIDRef;
import de.lmu.ifi.dbs.elki.database.ids.DoubleDBIDListIter;
import de.lmu.ifi.dbs.elki.database.ids.KNNList;
import de.lmu.ifi.dbs.elki.parallel.Executor;
import de.lmu.ifi.dbs.elki.parallel.processor.AbstractDoubleProcessor;
import de.lmu.ifi.dbs.elki.parallel.processor.KNNProcessor;
import de.lmu.ifi.dbs.elki.parallel.variables.SharedDouble;
import de.lmu.ifi.dbs.elki.parallel.variables.SharedObject;
/**
* Compute the kNN weight score, used by {@link ParallelKNNWeightOutlier}.
*
* Needs the k nearest neighbors as input, for example from {@link KNNProcessor}
*
* @author Erich Schubert
*/
public class KNNWeightProcessor extends AbstractDoubleProcessor {
/**
* K parameter
*/
int k;
/**
* Constructor.
*
* @param k K parameter
*/
public KNNWeightProcessor(int k) {
super();
this.k = k;
}
/**
* KNN query object
*/
SharedObject<? extends KNNList> input;
/**
* Connect the input channel.
*
* @param input Input channel
*/
public void connectKNNInput(SharedObject<? extends KNNList> input) {
this.input = input;
}
@Override
public Instance instantiate(Executor executor) {
return new Instance(k, executor.getInstance(input), executor.getInstance(output));
}
/**
* Instance for precomputing the kNN.
*
* @author Erich Schubert
*
* @apiviz.exclude
*/
private static class Instance extends AbstractDoubleProcessor.Instance {
/**
* k Parameter
*/
int k;
/**
* kNN query
*/
SharedObject.Instance<? extends KNNList> input;
/**
* Constructor.
*
* @param k K parameter
* @param input kNN list input
* @param store Datastore to write to
*/
protected Instance(int k, SharedObject.Instance<? extends KNNList> input, SharedDouble.Instance store) {
super(store);
this.k = k;
this.input = input;
}
@Override
public void map(DBIDRef id) {
final KNNList list = input.get();
int i = 0;
double sum = 0;
for(DoubleDBIDListIter iter = list.iter(); iter.valid() && i < k; iter.advance(), ++i) {
sum += iter.doubleValue();
}
output.set(sum);
}
}
}
|