When running the quick_start_inference.ipynb notebook on the TCGA-UCS dataset, a KeyError occurs during the feature loading stage. The script attempts to access a 'labels' key within the .h5 file that does not appear to exist in the current feature files extracted via the KEEP encoder.
Could you please advise if I might have missed a specific step in the data preparation, or if there is a recommended way to resolve this? I would truly appreciate your guidance on this matter. Thank you for your time and for sharing this excellent work!