myogestic.models.definitions.raulnet_models.train#

myogestic.models.definitions.raulnet_models.train(model, dataset, _, gui_logger)[source]#

Train a RaulNet model using preprocessed features from the dataset.

Parameters:
  • model (L.LightningModule) – The RaulNet model to train.

  • dataset (dict) – The dataset containing preprocessed EMG features and kinematics. Expected keys: - “emg”: np.ndarray with shape (n_samples, n_features * channels, time) - “kinematics”: np.ndarray with shape (n_samples, n_outputs) - “buffer_size__samples”: int (used for model configuration)

  • gui_logger (CustomLogger) – Logger for outputting training progress to the GUI.

  • _ (bool)

Returns:

The trained RaulNet model.

Return type:

L.LightningModule