import numpy as np
import os

def check_data(name, dir_path):
    print(f"--- Checking {name} ---")
    x_path = os.path.join(dir_path, f"X_{name}.npy")
    y_path = os.path.join(dir_path, f"y_{name}.npy")
    
    if os.path.exists(x_path):
        x = np.load(x_path)
        y = np.load(y_path)
        print(f"X shape: {x.shape}")
        print(f"y shape: {y.shape}")
        unique, counts = np.unique(y, return_counts=True)
        print(f"Classes: {dict(zip(unique, counts))}")
    else:
        print(f"File not found: {x_path}")

base = r"c:\Users\ASUS\lung_ai_project\data"
check_data("hear", os.path.join(base, "hear_embeddings"))
check_data("hear_aug", os.path.join(base, "hear_embeddings_augmented"))