test.py

import pandas as pd

def check_data(partition_id):
    # Load the correct dataset based on the partition_id
    if partition_id == 0:
        data = pd.read_csv("hh101.ann.features.csv")
    else:
        data = pd.read_csv("hh102.ann.features.csv")

    # Check the unique values in the 'activity' column
    unique_activities = data['activity'].unique()
    num_classes = len(unique_activities)
    
    print(f"Unique activity labels for partition {partition_id}: {unique_activities}")
    print(f"Number of unique classes for partition {partition_id}: {num_classes}")

if __name__ == "__main__":
    # Check the number of classes for both clients
    check_data(partition_id=0)  # Client 1
    check_data(partition_id=1)  # Client 2