ssitu / java-neural-network-library Goto Github PK

This library is seriously impressive, but like please don't use magic strings for parameters. For example:
String hiddenActivationFunction, String outputActivationFunction, String lossFunction, String optimizer from the constructor can be changed to enums of the same name. So in the end it would be in the constructor: HiddenActivationFunction name, OutputActivationFunction name, LossFunction name, Optimizer name (replace name obviously). The enum for Optimizer would be:

public enum LossFunction{ quadratic, log, crossEntropy }

Or something to that effect.