Skip to content

Commit 29bc323

Browse files
committed
Fix the bug with the output normalization in the classifier
1 parent 25ea87a commit 29bc323

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/main/java/com/github/chen0040/mlp/ann/MLP.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ public void train(DataFrame batch, int training_epoches)
7575

7676
double[] target = getTarget(row);
7777

78-
if (outputNormalization != null) {
78+
if (normalizeOutputs) {
7979
target = outputNormalization.standardize(target);
8080
}
8181

@@ -128,7 +128,7 @@ public void train(DataFrame batch, int training_epoches)
128128

129129
double[] target = getTarget(row);
130130

131-
if (outputNormalization != null) {
131+
if (normalizeOutputs) {
132132
target = outputNormalization.standardize(target);
133133
}
134134

src/test/java/com/github/chen0040/mlp/ann/regression/MLPRegressionUnitTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ public static double randn(){
4242
return r * Math.sin(theta);
4343
}
4444

45-
45+
4646

4747
@Test
4848
public void test_simple_regression() {

0 commit comments

Comments
 (0)