edu.upb.FacilitySupportBot/classificationEvaluationC45.js at master · geskill/edu.upb.FacilitySupportBot · GitHub

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
var C45 = require('c4.5');

var incidentClassification = require('./classification.json');

var features = ['attr1', 'attr2'];
var featureTypes = ['category', 'category'];

/*
 * Classification of type
 */

console.log('c4.5: type');

var trainingData = [];
var incidentTypeClassification = JSON.parse(JSON.stringify(incidentClassification)); // deep clone hack
for (let set of incidentTypeClassification.trainset) {
	var o = [];
	o.push(set.input.reference || set.input.position);
	o.push(set.input.symptom);
	o.push(set.output.type);

	trainingData.push(o);
}

var target = 'type';
var c45 = C45();

c45.train({
	data: trainingData,
	target: target,
	features: features,
	featureTypes: featureTypes
}, function (error, model) {
	if (error) {
		console.error(error);
		return false;
	}

	var successCount = 0;
	var errorCount = 0;

	var testData = [];
	for (let set of incidentTypeClassification.testset) {
		var o = [];
		o.push(set.input.reference || set.input.position);
		o.push(set.input.symptom);
		o.push(set.output.type);

		var result = model.classify(o);

		if (result === set.output.type) {
			successCount++;
		} else {
			errorCount++;
		}

		// console.log(model.classify(o));

	}
	console.log('success: ' + successCount);
	console.log('error: ' + errorCount);
});


/*
 * Classification of priority
 */

console.log('c4.5: priority');

var trainingData = [];
var incidentTypeClassification = JSON.parse(JSON.stringify(incidentClassification)); // deep clone hack
for (let set of incidentTypeClassification.trainset) {
	var o = [];
	o.push(set.input.position || set.input.reference);
	o.push(set.input.symptom);
	o.push(set.output.priority);

	trainingData.push(o);
}

var target = 'type';
var c45 = C45();

c45.train({
	data: trainingData,
	target: target,
	features: features,
	featureTypes: featureTypes
}, function (error, model) {
	if (error) {
		console.error(error);
		return false;
	}

	var successCount = 0;
	var errorCount = 0;

	var testData = [];
	for (let set of incidentTypeClassification.testset) {
		var o = [];
		o.push(set.input.position || set.input.reference);
		o.push(set.input.symptom);
		o.push(set.output.priority);

		var result = model.classify(o);

		if (result === set.output.priority) {
			successCount++;
		} else {
			errorCount++;
		}

		// console.log(model.classify(o));

	}
	console.log('success: ' + successCount);
	console.log('error: ' + errorCount);
});