Prateek Joshi comments

Repositories
Issues
Comments

Results 4 comments of


                                            Prateek Joshi

Multi-label classification issue

Hi, you don't have to change the softmax function. Just replace the output units from 2 to the number of classes in your dataset in the layer below: `self.fc2 =...

Extend for multi class classification

Hi, simply replace the output units from 2 to the number of classes in your dataset in the layer below: self.fc2 = nn.Linear(512,2)

Unfreezing Bert Parameters and training only on BERT

Hi, if you are not using the below code then it means that all the parameters of the BERT architecture will get tuned during model training. ``` for param in...

Questions about Multi-Label and output activation function...

Hi @JessicaKuo May I know which dataset you are using for multi-label classification?