| import numpy as np | |
| from sklearn.utils.class_weight import compute_class_weight | |
| def get_class_weight_dict(classes: list[str], y: np.ndarray) -> dict[str, float]: | |
| weights = compute_class_weight(class_weight="balanced", classes=classes, y=y) | |
| class_weight_dict = dict(enumerate(weights)) | |
| return class_weight_dict | |