francois/HowBadIsMyBatch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
frankknoll
2023-11-17 08:16:01 +01:00
parent 62ce64308c
commit 2b81552d7a
1 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/SymptomsCausedByVaccines/MultiLineFitting/MultiLineFitter.py
View File

@@ -29,15 +29,14 @@ class MultiLineFitter:
bestClusterIndexCombination = None
keepClustering = False
numClusters = preferenceMatrix.shape[0]
for clusterIndexA in range(numClusters):
for (clusterIndexA, clusterIndexB) in MultiLineFitter._getPairs(numClusters):
preferenceSetA = preferenceMatrix[clusterIndexA]
for clusterIndexB in range(clusterIndexA):
preferenceSetB = preferenceMatrix[clusterIndexB]
similarity = MultiLineFitter._intersectionOverUnion(preferenceSetA, preferenceSetB);
if similarity > maxSimilarity:
keepClustering = True
maxSimilarity = similarity
bestClusterIndexCombination = (clusterIndexA, clusterIndexB)
preferenceSetB = preferenceMatrix[clusterIndexB]
similarity = MultiLineFitter._intersectionOverUnion(preferenceSetA, preferenceSetB);
if similarity > maxSimilarity:
keepClustering = True
maxSimilarity = similarity
bestClusterIndexCombination = (clusterIndexA, clusterIndexB)
if keepClustering:
(clusterIndexA, clusterIndexB) = bestClusterIndexCombination
@@ -48,6 +47,12 @@ class MultiLineFitter:
return clusters, preferenceMatrix
@staticmethod
def _getPairs(n):
for i in range(n):
for j in range(i):
yield (i, j)
@staticmethod
def _intersectionOverUnion(setA, setB):
intersection = np.count_nonzero(np.logical_and(setA, setB))