Merge pull request #5 from piotrgrudzien/master

Numeric underflow causing NaNs in the output
karpathy · Oct 24, 2016 · 13ece5d · 13ece5d
2 parents 2d62a1c + 6921834
commit 13ece5d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.DS_Store b/.DS_Store
diff --git a/tsne.js b/tsne.js
@@ -130,7 +130,11 @@ var tsnejs = tsnejs || { REVISION: 'ALPHA' };
  // normalize p and compute entropy
  var Hhere = 0.0;
  for(var j=0;j<N;j++) {
- var pj = prow[j] / psum;
+ if(psum == 0) {
+ var pj = 0;
+ } else {
+ var pj = prow[j] / psum;
+ }
  prow[j] = pj;
  if(pj > 1e-7) Hhere -= pj * Math.log(pj);
  }