Fixed potential 0/0 errors causing NaN in the output

karpathy · Oct 22, 2016 · 4be77e8 · 4be77e8
1 parent 2d62a1c
commit 4be77e8
Showing 1 changed file with 5 additions and 1 deletion.
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);
  }