import numpy as np

def check_shapes():
    try:
        # Load the saved numpy arrays
        predictions = np.load("predictions_1mil.npy")
        coordinates = np.load("coordinates_1mil.npy")
        
        # Print their shapes
        print("Predictions shape:", predictions.shape)
        print("Coordinates shape:", coordinates.shape)
        
        # Print total number of elements for comparison
        print("Number of predictions:", predictions.size)
        print("Number of coordinate pairs:", coordinates.shape[0])
        
    except FileNotFoundError as e:
        print(f"Error: One or both files not found - {e}")
    except Exception as e:
        print(f"Error loading files: {e}")

if __name__ == "__main__":
    check_shapes()