Package org.apache.solr.ltr.model
Class NeuralNetworkModel.DefaultLayer
- java.lang.Object
-
- org.apache.solr.ltr.model.NeuralNetworkModel.DefaultLayer
-
- All Implemented Interfaces:
NeuralNetworkModel.Layer
- Enclosing class:
- NeuralNetworkModel
public class NeuralNetworkModel.DefaultLayer extends Object implements NeuralNetworkModel.Layer
-
-
Field Summary
Fields Modifier and Type Field Description protected NeuralNetworkModel.Activation
activation
protected String
activationStr
-
Constructor Summary
Constructors Constructor Description DefaultLayer()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description float[]
calculateOutput(float[] inputVec)
String
describe()
void
setActivation(Object activationStr)
void
setBias(Object biasObj)
void
setMatrix(Object matrixObj)
int
validate(int inputDim)
-
-
-
Field Detail
-
activationStr
protected String activationStr
-
activation
protected NeuralNetworkModel.Activation activation
-
-
Method Detail
-
setMatrix
public void setMatrix(Object matrixObj)
-
setBias
public void setBias(Object biasObj)
-
setActivation
public void setActivation(Object activationStr)
-
calculateOutput
public float[] calculateOutput(float[] inputVec)
- Specified by:
calculateOutput
in interfaceNeuralNetworkModel.Layer
-
validate
public int validate(int inputDim) throws ModelException
- Specified by:
validate
in interfaceNeuralNetworkModel.Layer
- Throws:
ModelException
-
describe
public String describe()
- Specified by:
describe
in interfaceNeuralNetworkModel.Layer
-
-