public class LogisticRegression {
private final SmallLayeredNeuralNetwork ann;
public LogisticRegression(int dimension) {
this.ann = new SmallLayeredNeuralNetwork();
this.ann.addLayer(dimension, false, FunctionFactory.createDoubleFunction("Sigmoid"));
this.ann.addLayer(1, true, FunctionFactory.createDoubleFunction("Sigmoid"));
this.ann.setCostFunction(FunctionFactory.createDoubleDoubleFunction("CrossEntropy"));
}