PyTestforDataScience_PyDataLA icon indicating copy to clipboard operation
PyTestforDataScience_PyDataLA copied to clipboard

Published 20 hours ago • verified publisher icon canyon289

Implementing unit tests for pytorch models

Open anidiatm41 opened this issue 2 years ago • 0 comments

Hi Ravin, Thanks for educating us in py test. I have a NER model trained and saved in my local in .pkl format. I use that model for prediction (Its a pytorch model). Now I want to implement the unit test cases for the prediction function using mock . How can I do it?

anidiatm41 avatar Feb 06 '23 04:02 anidiatm41