refactoring

This commit is contained in:
frankknoll
2023-11-17 08:01:08 +01:00
parent fcd8614420
commit 62ce64308c

View File

@@ -25,7 +25,7 @@ class MultiLineFitter:
numClusters = preferenceMatrix.shape[0] numClusters = preferenceMatrix.shape[0]
clusters = [[i] for i in range(numClusters)] clusters = [[i] for i in range(numClusters)]
while keepClustering: while keepClustering:
maxDistance = 0 maxSimilarity = 0
bestClusterIndexCombination = None bestClusterIndexCombination = None
keepClustering = False keepClustering = False
numClusters = preferenceMatrix.shape[0] numClusters = preferenceMatrix.shape[0]
@@ -33,10 +33,10 @@ class MultiLineFitter:
preferenceSetA = preferenceMatrix[clusterIndexA] preferenceSetA = preferenceMatrix[clusterIndexA]
for clusterIndexB in range(clusterIndexA): for clusterIndexB in range(clusterIndexA):
preferenceSetB = preferenceMatrix[clusterIndexB] preferenceSetB = preferenceMatrix[clusterIndexB]
distance = MultiLineFitter._intersectionOverUnion(preferenceSetA, preferenceSetB); similarity = MultiLineFitter._intersectionOverUnion(preferenceSetA, preferenceSetB);
if distance > maxDistance: if similarity > maxSimilarity:
keepClustering = True keepClustering = True
maxDistance = distance maxSimilarity = similarity
bestClusterIndexCombination = (clusterIndexA, clusterIndexB) bestClusterIndexCombination = (clusterIndexA, clusterIndexB)
if keepClustering: if keepClustering: