search

krishnaik06 / Huggingfacetransformer

GNU General Public License v3.0
59 stars 97 forks source link

Fixed bug in prediction and CNF, which gave 100% accuracy #3

Open prabathbr opened 2 years ago

prabathbr commented 2 years ago

Changes :

output=trainer.predict(test_dataset)[0]

import numpy as np output = np.argmax(output, axis = - 1)