Skip to content

m3cv KeyError: 'SubjectID' from running prepare_pretrain_dataset.py #13

@ChiShengChen

Description

@ChiShengChen

Traceback (most recent call last):
File "/home/aidan/EEGPT/datasets/pretrain/prepare_pretrain_dataset.py", line 244, in
dataset = get_M3CV_dataset()
File "/home/aidan/EEGPT/datasets/pretrain/prepare_pretrain_dataset.py", line 200, in get_M3CV_dataset
dataset = M3CVDataset(
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/torcheeg/datasets/module/personal_identification/m3cv.py", line 154, in init
super().init(**params)
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/torcheeg/datasets/module/base_dataset.py", line 71, in init
raise e
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/torcheeg/datasets/module/base_dataset.py", line 60, in init
self.save_record(io_path=self.io_path,
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/torcheeg/datasets/module/base_dataset.py", line 228, in save_record
obj = next(gen)
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/torcheeg/datasets/module/personal_identification/m3cv.py", line 181, in process_record
'subject_id': epoch_info['SubjectID'],
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/pandas/core/series.py", line 981, in getitem
return self._get_value(key)
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/pandas/core/series.py", line 1089, in _get_value
loc = self.index.get_loc(label)
File "/home/aidan/anaconda3/envs/torcheeg/lib/python3.10/site-packages/pandas/core/indexes/base.py", line 3804, in get_loc
raise KeyError(key) from err
KeyError: 'SubjectID'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions