File size: 753 Bytes
a16f583 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | 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()
|