Commit History

Increase threshold
99575b1

jesse-tong commited on

Edit evaluation to use weighted instead as there are imbalance labels
bad72d7

jesse-tong commited on

Fix a bug in evaluation in multi-class
80b78df

jesse-tong commited on

Forget num_classes
6cf4c1f

jesse-tong commited on

Change loss function for multi-category because CrossEntropyLoss only accept outputs of (batch, x) and labels with shape of (batch)
cb428cb

jesse-tong commited on