FilesExpand file tree

master

/

ClassificationVoter.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

74 lines (53 loc) · 1.88 KB

master

/

ClassificationVoter.java

Copy path

Top

File metadata and controls

• Code
• Blame

74 lines (53 loc) · 1.88 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package utils;

import java.util.*;

public class ClassificationVoter implements IVoter<Integer, Double, List<Pair<Integer, Double>> > {

public ClassificationVoter(){

this.values = new ArrayList<Pair<Integer, Double>>();

}

@Override

public void addItem(Integer item, Double criterionType){

this.values.add(PairBuilder.makePair(item, criterionType));

}

/**

* Allow any Object cast to double to be added

* @param item

* @param criterionType

*/

public void addItem(Integer item, Object criterionType){

this.addItem(item, (Double) criterionType);

}

/**

* Add a list of items T, U should be castable to Integer and Double respectively

*/

public <T, U> void addItems(List<Pair<T, U>> items){

for (int i = 0; i < items.size(); i++) {

this.values.add((Pair<Integer, Double>)items.get(i));

}

}

@Override

public List<Pair<Integer, Double>> getResult(int max){

//get the values

Object[] valuesArr = (Object[]) values.toArray();

Arrays.sort(valuesArr, new Comparator<Object>() {

@Override

public int compare(Object t, Object t1) {

if(((Pair<Integer, Double>)t).second < ((Pair<Integer, Double>)t1).second ){

return -1;

}

else if(((Pair<Integer, Double>)t).second > ((Pair<Integer, Double>)t1).second){

return 1;

}

return 0;

}

});

List<Pair<Integer, Double>> subList = new ArrayList<Pair<Integer, Double>>();

for(int i=0; i<max; ++i){

subList.add((Pair<Integer, Double>) valuesArr[i]);

}

return subList;

}

public void clear(){

this.values = new ArrayList<Pair<Integer, Double>>();

}

private List<Pair<Integer, Double>> values;

}