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
|
"""!
@brief pyclustering module for samples.
@authors Andrei Novikov (pyclustering@yandex.ru)
@date 2014-2020
@copyright BSD-3-Clause
"""
class answer_reader:
"""!
@brief Answer reader for samples that are used by pyclustering library.
"""
def __init__(self, answer_path):
"""!
@brief Creates instance of answer reader to read proper clustering results of samples.
@param[in] answer_path (string): Path to clustering results (answers).
"""
self.__answer_path = answer_path
self.__clusters = None
self.__noise = None
def get_clusters(self):
"""!
@brief Read proper clustering results.
@return (list) Clusters where each cluster is represented by list of index point from dataset.
"""
self.__read_answer()
return self.__clusters
def get_noise(self):
"""!
@brief Read proper clustering results
@return (list) Noise where each outlier is represented by index point from dataset.
"""
self.__read_answer()
return self.__noise
def get_cluster_lengths(self):
"""!
@brief Read proper cluster lengths.
@details Cluster length means amount of point in a cluster.
@return (list) Cluster lengths where each length means amount of points in a cluster.
"""
clusters = self.get_clusters()
return [len(cluster) for cluster in clusters]
def __read_answer_from_line(self, index_point, line):
"""!
@brief Read information about point from the specific line and place it to cluster or noise in line with that
information.
@param[in] index_point (uint): Index point that should be placed to cluster or noise.
@param[in] line (string): Line where information about point should be read.
"""
if line[0] == 'n':
self.__noise.append(index_point)
else:
index_cluster = int(line)
if index_cluster >= len(self.__clusters):
self.__clusters.append([index_point])
else:
self.__clusters[index_cluster].append(index_point)
def __read_answer(self):
"""!
@brief Read information about proper clusters and noises from the file.
"""
if self.__clusters is not None:
return
file = open(self.__answer_path, 'r')
self.__clusters, self.__noise = [], []
index_point = 0
for line in file:
self.__read_answer_from_line(index_point, line)
index_point += 1
file.close()
|
