gist-ailab / SleePyCo

This is an official implementation for "SleePyCo: Automatic Sleep Scoring with Feature Pyramid and Contrastive Learning".
MIT License
35 stars 11 forks source link

I am having a new issue #7

Closed Brandoncheaa closed 2 months ago

Brandoncheaa commented 2 months ago

Originally I had the code working, I am now running into this issue when i am now evaluating the dataset using the command provided: envs/sleepyco/lib/python3.8/site-packages/sklearn/metrics/_classification.py:697: RuntimeWarning: invalid value encountered in true_divide k = np.sum(w_mat confusion) / np.sum(w_mat expected) Traceback (most recent call last): File "test.py", line 89, in main() File "test.py", line 85, in main summarize_result(config, fold, Y_true, Y_pred) File "/cluster/academic/projects/ASCS/env/SleePyCo/SleePyCo/utils.py", line 184, in summarize_result wpr = round(result_dict['0.0']['precision']*100, 1) KeyError: '0.0'