ca14cf72e3e4bd3a6136cafad8701ac41f48f09b,yellowbrick/cluster/elbow.py,KElbowVisualizer,__init__,#KElbowVisualizer#,206

Before Change

            k = (2, k+1)

        // Expand k in to the values we will use, capturing exceptions
        try:
            k = tuple(k)
            self.k_values_ = list(range(*k))
        except:
            raise YellowbrickValueError((
                "Specify a range or maximal K value, the value "{}" "
                "is not a valid argument for K.".format(k)
            ))


        // Holds the values of the silhoutte scores
        self.k_scores_ = None

    def fit(self, X, y=None, **kwargs):
        

After Change

            self.k_values_ = list(range(*k))
        elif isinstance(k, collections.Iterable) and \
                all(isinstance(x, (int, np.integer)) for x in k):
            self.k_values_ = list(k)
        else:
            raise YellowbrickValueError((
                "Specify an iterable of integers, a range, or maximal K value,"
                " the value "{}" is not a valid argument for K.".format(k)

Instances