FilesExpand file tree

master

/

ConfusionMatrix.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

79 lines (54 loc) · 1.82 KB

master

/

ConfusionMatrix.java

Copy path

Top

File metadata and controls

• Code
• Blame

79 lines (54 loc) · 1.82 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

75

76

77

78

79

package maths;

import datastructs.maths.DenseMatrixSet;

import datastructs.maths.RowBuilder;

import datastructs.utils.RowType;

import utils.Pair;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

/**

* ConfusionMatrix

*/

public class ConfusionMatrix {

public ConfusionMatrix(List<Pair<Integer, Integer>> classes, List<String> names){

if(classes == null){

throw new IllegalArgumentException("Classes should not be null");

}

this.names = names;

data_ = new DenseMatrixSet<>(RowType.Type.INTEGER_VECTOR, new RowBuilder(), names.size(), names.size(), -1);

Map<Integer, List<Integer>> result = new HashMap<>();

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

Pair<Integer, Integer> pair = classes.get(i);

Integer classIdx = pair.first;

if(result.containsKey(classIdx)){

result.get(classIdx).add(classIdx, 1);

}

else{

List<Integer> values = new ArrayList<>(names.size());

for(int j=0; j<names.size(); ++j){

values.add(0);

}

result.put(classIdx, values);

}

}

}

@Override

public String toString(){

StringBuilder builder = new StringBuilder();

builder.append(" | ");

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

builder.append(names.get(i)+ " | ");

}

builder.append("\n");

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

for(int l=0; l<this.names.get(i).length(); ++l)

builder.append("_");

}

builder.append("\n");

builder.append(" | ");

return builder.toString();

}

private List<String> names;

private DenseMatrixSet<Integer> data_;

}