prateekjoshi565 / Fine-Tuning-BERT

Apache License 2.0
139 stars 119 forks source link

axis 1 is out of bounds for array of dimension 1 #16

Open yuanqino opened 1 year ago

yuanqino commented 1 year ago

I used my own dataset to train a Bert model but an issue occur when I am trying to evaluate the model with testing dataset (refer to the image 1). The shape of the array is (1909,2) which is a 2 dimensional array. One more issue is the original size of testing data should be 1910, not sure why the shape of preds is become 1909.

Image 1 image

Image 2 image