test.py

# import h5py

# def print_model_details(file_path):
#     with h5py.File(file_path, 'r') as f:
#         print(f.keys())  # Print layers
#         print(len(f.keys())) 
#         print("")
#         for key in f.keys():
#             print(len(list(f[key].keys())))
#             print(f"{key}: {list(f[key].keys())}")  # Print details of each layer)
# print('rock')
# print_model_details('model_classification/rock-170.h5')
# print('mummified-170')
# print_model_details('model_classification/mummified-170.h5')
# print('BEiT')
# print_model_details('model_classification/fossil-142.h5')
# print('BEiT New')
# print_model_details('model_classification/fossil-new.h5')
# print("Newest:")
# print_model_details('model_classification/fossil-model.h5')


import numpy as np

# Load the .npy file
embedding = np.load('dataset/embedding_leaves_142_finer.npy')

# Check the shape of the array
print(embedding.shape)