Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/compute_class_wt_bert.py

Created July 18, 2020 10:41
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Embed
What would you like to do?
Learn more about clone URLs
Download ZIP
Raw
compute_class_wt_bert.py
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