MichaelWelsch commited on
Commit
302b190
1 Parent(s): d984c08

Update funcs/som.py

Browse files
Files changed (1) hide show
  1. 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=1.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.
 
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.