NeuSpeech / EEG-To-Text

72 stars 8 forks source link

Update data.py #8

Closed Dark-knight-02 closed 1 week ago

Dark-knight-02 commented 1 week ago

There was an issue with the line sent_level_eeg_tensor = torch.from_numpy(sent_obj['sentence_level_rawEEG']) The sent_obj['sentence_level_rawEEG'] gives out a dictionary Also the data prepared using the construct_dataset_mat_to_pickle_v1.py makes the pickle file with sent_obj['sentence_level_EEG'] which was sent_obj['sentence_level_rawEEG'] in the code of data.py

Same with the sent_obj['word_ft'] and sent_obj['word'] keyword variation which caused an error.

NeuSpeech commented 1 week ago

Thanks for your contribution. Can you show me a screenshot or some evidence of successful running with your code?

Dark-knight-02 commented 1 week ago

Please find the screenshots.

Screenshot 2024-06-25 223856 Screenshot 2024-06-25 223922 Screenshot 2024-06-25 223950

NeuSpeech commented 1 week ago

Thank you!