e753bb1b0170eb3d6fabc93311c31f4094f1cdad,tslearn/tests/test_serialize_models.py,,test_serialize_global_alignment_kernel_kmeans,#,113

Before Change

    numpy.random.seed(seed)
    X_train, y_train, X_test, y_test = CachedDatasets().load_dataset("Trace")
    // Keep first 3 classes
    X_train = X_train[y_train < 4]
    numpy.random.shuffle(X_train)
    // Keep only 50 time series
    X_train = TimeSeriesScalerMeanVariance().fit_transform(X_train[:50])
    sz = X_train.shape[1]

    gak_km = GlobalAlignmentKernelKMeans(n_clusters=3,
                                         sigma=sigma_gak(X_train),
                                         n_init=1,

After Change

def test_serialize_global_alignment_kernel_kmeans():
    n, sz, d = 15, 10, 3
    rng = numpy.random.RandomState(0)
    X = rng.randn(n, sz, d)

    gak_km = GlobalAlignmentKernelKMeans(n_clusters=3, verbose=False,
                                         max_iter=5)

Instances