Class Weighting

.py

# Import class weights function
from sklearn.utils.class_weight import compute_class_weight

# Calculate class weights
class_weights = compute_class_weight('balanced', np.unique(y_train), y_train)

# Create a dictionary to map class indices to their respective weights
class_weight_dict = dict(zip(np.unique(y_train), class_weights))

# Define and train a classifier with class weights
classifier = RandomForestClassifier(class_weight=class_weight_dict)