Spaces:
Build error
Build error
MichaelWelsch
commited on
Commit
•
302b190
1
Parent(s):
d984c08
Update funcs/som.py
Browse files- funcs/som.py +1 -1
funcs/som.py
CHANGED
@@ -30,7 +30,7 @@ class ClusterSOM:
|
|
30 |
if len(model_data) > 5:
|
31 |
self.label_centroids, self.label_encodings = model_data[5:]
|
32 |
|
33 |
-
def predict(self, data, sigma_factor=
|
34 |
"""
|
35 |
Predict the cluster and BMU SOM coordinate for each sample in the data if it's inside the sigma value.
|
36 |
Also, predict the label and distance to the center of the label if labels are trained.
|
|
|
30 |
if len(model_data) > 5:
|
31 |
self.label_centroids, self.label_encodings = model_data[5:]
|
32 |
|
33 |
+
def predict(self, data, sigma_factor=2.5):
|
34 |
"""
|
35 |
Predict the cluster and BMU SOM coordinate for each sample in the data if it's inside the sigma value.
|
36 |
Also, predict the label and distance to the center of the label if labels are trained.
|