Skip to content

Commit f50af3f

Browse files
committed
Fixed bug in normalization from code refactoring.
1 parent a96cae9 commit f50af3f

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

lib/NN.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1187,7 +1187,7 @@ def driver(inputdir, outputdir, datadir, plotdir, preddir,
11871187
if normalize or scale:
11881188
if normalize:
11891189
print('\nNormalizing the data...', flush=True)
1190-
else:
1190+
if scale:
11911191
print('\nScaling the data...', flush=True)
11921192

11931193
try:
@@ -1229,6 +1229,11 @@ def driver(inputdir, outputdir, datadir, plotdir, preddir,
12291229
y_min = U.normalize(y_min, y_mean, y_std)
12301230
y_max = U.normalize(y_max, y_mean, y_std)
12311231
else:
1232+
x_mean = 0.
1233+
x_std = 1.
1234+
y_mean = 0.
1235+
y_std = 1.
1236+
if scale:
12321237
if np.any(x_min == x_max):
12331238
ibad = np.where(x_min == x_max)
12341239
raise ValueError("The minimum and maximum are equal at the " +\
@@ -1242,6 +1247,12 @@ def driver(inputdir, outputdir, datadir, plotdir, preddir,
12421247
if verb:
12431248
print("min:", x_min, y_min)
12441249
print("max:", x_max, y_max)
1250+
else:
1251+
x_min = 0.
1252+
x_max = 1.
1253+
y_min = 0.
1254+
y_max = 1.
1255+
scalelims = [0., 1.]
12451256
else:
12461257
x_mean = 0.
12471258
x_std = 1.

0 commit comments

Comments
 (0)