Multi-class classification - PyTorch Forums
discuss.pytorch.org › t › multi-class-classificationJun 10, 2019 · I am trying to do a multi-class classification in pytorch. The code runs fine, but the accuracy is not good. I was wondering if my code is correct? The input to the model is a matrix of 2000x100 and the output is a 1D tensor with the index of the label ex: tensor([2,5,31,…,7]) => 2000 elements # another multi-class classification class MultiClass(nn.Module): def __init__(self, x_dim, z_dim ...