In [1]:
import dwdii_transforms

In [2]:
dataPath = "/root/facial_expressions/data/legend.csv"
metaData = dwdii_transforms.load_training_metadata(dataPath, balanceViaRemoval=True, verbose=True)


Before Balancing
----------------
sadness 268
neutral 6868
contempt 9
disgust 208
anger 252
surprise 368
fear 21
happiness 5696

After Balancing
----------------
sadness 268
neutral 568
contempt 9
disgust 208
anger 252
surprise 368
fear 21
happiness 568

In [3]:
rawCount = 268 + 6868 + 9 + 208 + 252 + 368 + 21 + 5696
print rawCount


13690

In [4]:
6868.0 / rawCount


Out[4]:
0.5016800584368152

In [5]:
5696.0 / rawCount


Out[5]:
0.4160701241782323

In [6]:
balancedCount = 268 + 568 + 9 + 208 + 252 + 368 + 21 + 568
print balancedCount


2262

In [ ]: