File size: 1,054 Bytes
dec6e74 a181709 dec6e74 a181709 dec6e74 a181709 dec6e74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
---
language: en
tags:
- eeg
- meg
- pytorch
- neuroimaging
license: mit
datasets:
- gabrycina/eeg2meg-tiny
metrics:
- mse
---
# EEG to MEG Prediction Model
This model was trained to predict MEG signals from EEG recordings.
## Training Configuration
- Dataset: gabrycina/eeg2meg-tiny
- Batch Size: 32
- Learning Rate: 0.0001
- Device: mps
- Training Date: 20250104_185119
## Performance
- Best Validation Loss: 0.171059
- Best Epoch: 100
## Model Description
This model uses a deep learning architecture to predict MEG signals from EEG recordings. The architecture includes:
- Frequency and temporal convolutions for feature extraction
- Multi-head attention mechanisms for sensor relationships
- Residual connections for better gradient flow
- Separate prediction heads for magnetometers and gradiometers
## Usage
```python
import torch
# Load the model
model = torch.load('best_model.pth')
# Prepare your EEG data (shape: [batch_size, channels, time_points])
# Make predictions
with torch.no_grad():
meg_predictions = model(eeg_data)
```
|