Source code for scitex_ml.utils._format_samples_for_sktime

import numpy as np
import pandas as pd
import torch


def _format_a_sample_for_sktime(x):
    """
    x.shape: (n_chs, seq_len)
    """
    dims = pd.Series(
        [pd.Series(x[d], name=f"dim_{d}") for d in range(len(x))],
        index=[f"dim_{i}" for i in np.arange(len(x))],
    )
    return dims




def format_samples_for_sktime(X):
    """
    X.shape: (n_samples, n_chs, seq_len)
    """
    if torch.is_tensor(X):
        X = X.numpy()  # (64, 160, 1024)

        X = X.astype(np.float64)

    return pd.DataFrame([_format_a_sample_for_sktime(X[i]) for i in range(len(X))])