-
Notifications
You must be signed in to change notification settings - Fork 211
I tried you code in mnist dataset, but I got NaN error #34
Comments
Traceback (most recent call last): |
I guess the reason lies in the version of sklearn, or pytorch, But I don't know the exact version you used, waiting for your answer! Could you please point out the reason? |
have you run the code successufully? and how about the result? |
I have run successfully, the result differs from the article, I got following AUC: 0.7897 , 0.4115 , 0.7857 , 0.6432 , 0.6759 , 0.7587, 0.745 , 0.5428 , 0.8484 , 0.5741 I modified the train step in code, because if not, it does not do any training, thus got NaN error. I make train step equal to ratio of number of samples and batch size per epoch. |
sorry,I don't run this code.and does the result be similar with the test image? |
Could you provide your changed code? |
I find this code not stable, then I tried this code: |
Thank you for your effective reply. |
No description provided.
The text was updated successfully, but these errors were encountered: