Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
from sklearn.utils.class_weight import compute_class_weight
#compute the class weights
class_weights = compute_class_weight('balanced', np.unique(train_labels), train_labels)
print("Class Weights:",class_weights)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment