com.expleague.ml.models.nn.layers.OutputLayerBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmll

Various ML methods implemented by myself and my students

There is a newer version: 1.4.9

package com.expleague.ml.models.nn.layers;

import com.expleague.commons.math.vectors.Vec;

public interface OutputLayerBuilder extends LayerBuilder {
  OutputLayer getLayer();
  OutputLayer build();

  @Override
  default LayerBuilder yStart(int yStart) {
    return this;
  }

  @Override
  default LayerBuilder wStart(int wStart) {
    return this;
  }


  interface OutputLayer extends Layer {
    Vec fromState(Vec state);

    @Override
    default int ydim() {
      return xdim();
    }

    @Override
    default int wdim() {
      return 0;
    }

    @Override
    default int yStart() {
      return -1;
    }

    @Override
    default void initWeights(Vec weights) { }
  }
}