به جای استفاده از این خط کد:

weights = compute_class_weight('balanced', np.unique(train_data.classes), train_data.classes)

از خط کد زیر استفاده کنید:

weights = compute_class_weight('balanced', classes = np.unique(train_data.classes), y= train_data.classes)