2ab1119596b420cc696a783271d617562ccf7e28,cesium/tests/fixtures.py,,sample_featureset,#,42

Before Change

                                           for i in range(n_channels)])
                 for f in features}
    fset = xr.Dataset(feat_dict)
    fset.coords["name"] = ts_names
    fset.coords["channel"] = range(n_channels)
    if targets:
        ts_targets = np.array(list(islice(cycle(targets), size)))

After Change

def sample_featureset(size, n_channels=1, features=["mean"], targets=None,
                      names=None, meta_features=[]):
    ts_names = np.arange(size).astype("str")
    index = pd.MultiIndex.from_tuples([(f, i) for f in features for i in range(n_channels)],
                                      names=["feature", "channel"])
    fset = pd.DataFrame(np.random.random((size, len(features) * n_channels)),
                        columns=index)
    if targets:
        targets = np.array(list(islice(cycle(targets), size)))
    if names:
        fset.index = names

Instances