search

cs230-stanford / cs230-code-examples

Code examples in pyTorch and Tensorflow for CS230
Other
3.79k stars 984 forks source link

The NLP NER example only predicts correct Os #12

Open rohun-tripathi opened 6 years ago

rohun-tripathi commented 6 years ago

Hi, the NER example has an accuracy of 0.82 or 0.78 after running for 10 epochs on the toy 10 values dataset. I was using this code to startup a NER task I have to do. It is just predicting 'O's. I checked on set of predictions at the end of the epochs and the I's and B's predicted were all wrong.

Use a score like F1 on the B/I to get a better idea.

Note - I am not a student in the course. and I know it is supposed to be starter code, but the way it is presented, one expects that it'll work for the toy example.

suragnair commented 6 years ago

Is this the tensorflow or the pytorch version?

rohun-tripathi commented 6 years ago

The PyTorch version