From 866332c37fee03ac8d14adc7cac4650d340e079e Mon Sep 17 00:00:00 2001 From: Joaquin Torres Bravo Date: Fri, 14 Jun 2024 13:31:23 +0200 Subject: [PATCH] Aux script to print shapes of matrices --- explicability/print_shapes.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 explicability/print_shapes.py diff --git a/explicability/print_shapes.py b/explicability/print_shapes.py new file mode 100644 index 0000000..91a6342 --- /dev/null +++ b/explicability/print_shapes.py @@ -0,0 +1,15 @@ +import os +import numpy as np + +def print_shapes(directory): + # List all files in the directory + for file in os.listdir(directory): + if file.endswith(".npy"): + # Load the .npy file + data = np.load(os.path.join(directory, file)) + # Print the shape of the numpy array + print(f"{file}: {data.shape}") + +# Example usage: +directory_path = './output/shap_inter_values' +print_shapes(directory_path) \ No newline at end of file -- 2.24.1