contraction_net.prediction

Attributes

device

Functions

predict_contractions(data, model[, network])

predict contraction intervals time-series with neural network

Module Contents

contraction_net.prediction.device
contraction_net.prediction.predict_contractions(data, model, network=ContractionNet)[source]

predict contraction intervals time-series with neural network

Parameters:

  • data (ndarray) – 1D array with time-series of contraction

  • model (str) – trained model weights (.pt file)

  • network (nn.Module) – Network to predict contractions from time-series

  • standard_normalizer (bool, ndarray) – If False, each data is normalized by its mean and std. If True, the mean and std from the training data set are applied. If ndarray, the data is normalized with the entered args [[input_mean, input_std], [vel_mean, vel_std]]