I guess I am not able to find the target class

Also the following code is working for loading train data -

train_path = "/content/drive/MyDrive/train_data.npy"

train_data = np.load(train_path,allow_pickle=True)

train_data.shape

Can anyone guide as to where the target class is?? or do rectify the input file structure so that we can use the below code snippet for loading data as mentioned in the starter notebook

# Path to your HDF5 file

hdf5_file = "train_data.h5"

# Open the HDF5 file

with h5py.File(hdf5_file, 'r') as hdf:

# Extract the images (X)

X = np.array(hdf['images'])

# Extract the labels (y)

y = np.array(hdf['labels'])

# Check the shapes to ensure they are correct

print("Shape of X (images):", X.shape)

print("Shape of y (labels):", y.shape)

hdf5_file_test = "test_data.h5"

# Open the HDF5 file

with h5py.File(hdf5_file_test, 'r') as hdf:

# Extract the images (X)

X_test = np.array(hdf['images'])

# Check the shapes to ensure they are correct

print("Shape of X_test (images):", X_test.shape)