Skip to content

Commit 9486fb1

Browse files
committed
Issue #162: Use training constants to specify training type.
1 parent a71e1bc commit 9486fb1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/main/java/org/encog/app/analyst/wizard/AnalystWizard.java

+3-3
Original file line numberDiff line numberDiff line change
@@ -571,7 +571,7 @@ private void generateFeedForward(final int inputColumns,
571571
}
572572

573573
this.script.getProperties().setProperty(ScriptProperties.ML_TRAIN_TYPE,
574-
"rprop");
574+
MLTrainFactory.TYPE_RPROP);
575575
this.script.getProperties().setProperty(
576576
ScriptProperties.ML_TRAIN_TARGET_ERROR, this.maxError);
577577
}
@@ -983,10 +983,10 @@ private void generateRBF(final int inputColumns, final int outputColumns) {
983983

984984
if (outputColumns > 1) {
985985
this.script.getProperties().setProperty(
986-
ScriptProperties.ML_TRAIN_TYPE, "rprop");
986+
ScriptProperties.ML_TRAIN_TYPE, MLTrainFactory.TYPE_RPROP);
987987
} else {
988988
this.script.getProperties().setProperty(
989-
ScriptProperties.ML_TRAIN_TYPE, "svd");
989+
ScriptProperties.ML_TRAIN_TYPE, MLTrainFactory.TYPE_SVD);
990990
}
991991

992992
this.script.getProperties().setProperty(

0 commit comments

Comments
 (0)