package geniusweb.blingbling.Ranknet;


import java.util.LinkedList;
import java.util.List;

public class RanknetBuilder {
    private List<Layer> layers;
    private double learningRate = 1.0f;

    public RanknetBuilder() {
        this.layers = new LinkedList<Layer>();
    }

    public RanknetBuilder addLayer(Layer layer) {
        layers.add(layer);

        return this;
    }

    public RanknetBuilder setLearningRate(double rate) {
        this.learningRate = rate;

        return this;
    }

    public NeuralRankNet build() {
        NeuralRankNet neuralNet = new NeuralRankNet();
        neuralNet.setLayers(layers);
        neuralNet.setLearningRate(learningRate);

        return neuralNet;
    }
}