Pass kwargs to fit method in KElbowVisualizer

DistrictDataLabs · Apr 11, 2020 · d26b5ce · d26b5ce
1 parent 4737f0f
commit d26b5ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yellowbrick/cluster/elbow.py b/yellowbrick/cluster/elbow.py
@@ -309,7 +309,7 @@ def fit(self, X, y=None, **kwargs):
 
             # Set the k value and fit the model
             self.estimator.set_params(n_clusters=k)
-            self.estimator.fit(X)
+            self.estimator.fit(X, **kwargs)
 
             # Append the time and score to our plottable metrics
             self.k_timers_.append(time.time() - start)