Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/compute_class_wt_bert.py

Created July 18, 2020 10:41
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prateekjoshi565/f2526a6f5957ea3e56b39b475e1dcb76 to your computer and use it in GitHub Desktop.
Save prateekjoshi565/f2526a6f5957ea3e56b39b475e1dcb76 to your computer and use it in GitHub Desktop.
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