Skip to content

Commit 36ae8f2

Browse files
authored
Merge pull request #168 from amcadmus/devel
fix bug in descriptor nomalization
2 parents ed5be25 + 2b98a4e commit 36ae8f2

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
@@ -292,7 +292,10 @@ def _compute_dstats_sys_smth (self,
292292

293293

294294
def _compute_std (self,sumv2, sumv, sumn) :
295-
return np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
295+
val = np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
296+
if np.abs(val) < 1e-2:
297+
val = 1e-2
298+
return val
296299

297300

298301
def _filter(self,

source/train/DescrptSeR.py

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

241241

242242
def _compute_std (self,sumv2, sumv, sumn) :
243-
return np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
243+
val = np.sqrt(sumv2/sumn - np.multiply(sumv/sumn, sumv/sumn))
244+
if np.abs(val) < 1e-2:
245+
val = 1e-2
246+
return val
247+
244248

245249
def _filter_r(self,
246250
inputs,

0 commit comments

Comments
 (0)