Skip to content

Commit 07c42c1

Browse files
authored
Merge pull request #170 from amcadmus/master
fix bug in descriptor normalization
2 parents 88da716 + b29d6d3 commit 07c42c1

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

source/train/DescrptSeA.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,10 @@ def _compute_dstats_sys_smth (self,
289289

290290

291291
def _compute_std (self,sumv2, sumv, sumn) :
292-
return np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
292+
val = np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
293+
if np.abs(val) < 1e-2:
294+
val = 1e-2
295+
return val
293296

294297

295298
def _filter(self,

source/train/DescrptSeR.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,11 @@ def _compute_dstats_sys_se_r (self,
246246

247247

248248
def _compute_std (self,sumv2, sumv, sumn) :
249-
return np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
249+
val = np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
250+
if np.abs(val) < 1e-2:
251+
val = 1e-2
252+
return val
253+
250254

251255
def _filter_r(self,
252256
inputs,

0 commit comments

Comments
 (0)