Explicit passing of gamma paramater for kmeans (#1091)

Author: Alexsandruss

daal4py/sklearn/cluster/_k_means_0_22.py

@@ -104,23 +104,25 @@ def is_string(s, target_str):
 
 def _daal4py_kmeans_compatibility(nClusters, maxIterations, fptype="double",
                                   method="lloydDense", accuracyThreshold=0.0,
-                                  resultsToEvaluate="computeCentroids"):
+                                  resultsToEvaluate="computeCentroids", gamma=1.0):
     kmeans_algo = None
     if daal_check_version(((2020, 'P', 2), (2021, 'B', 107))):
         kmeans_algo = daal4py.kmeans(nClusters=nClusters,
                                      maxIterations=maxIterations,
                                      fptype=fptype,
                                      resultsToEvaluate=resultsToEvaluate,
                                      accuracyThreshold=accuracyThreshold,
-                                     method=method)
+                                     method=method,
+                                     gamma=gamma)
     else:
         assigFlag = 'computeAssignments' in resultsToEvaluate
         kmeans_algo = daal4py.kmeans(nClusters=nClusters,
                                      maxIterations=maxIterations,
                                      fptype=fptype,
                                      assignFlag=assigFlag,
                                      accuracyThreshold=accuracyThreshold,
-                                     method=method)
+                                     method=method,
+                                     gamma=gamma)
     return kmeans_algo
 
 

daal4py/sklearn/cluster/_k_means_0_23.py

@@ -139,14 +139,15 @@ def is_string(s, target_str):
 
 def _daal4py_kmeans_compatibility(nClusters, maxIterations, fptype="double",
                                   method="lloydDense", accuracyThreshold=0.0,
-                                  resultsToEvaluate="computeCentroids"):
+                                  resultsToEvaluate="computeCentroids", gamma=1.0):
     kmeans_algo = daal4py.kmeans(
         nClusters=nClusters,
         maxIterations=maxIterations,
         fptype=fptype,
         resultsToEvaluate=resultsToEvaluate,
         accuracyThreshold=accuracyThreshold,
         method=method,
+        gamma=gamma
     )
     return kmeans_algo