public class NNTrainedModel extends TrainedModel
Modifier and Type | Field and Description |
---|---|
private int |
numOfHidden |
private int |
numOfInputs |
private int |
numOfOutputs |
private float[][] |
Theta1 |
private float[][] |
Theta2 |
Constructor and Description |
---|
NNTrainedModel(int nInput,
int nHidden,
int nOutput,
float[] nn_params) |
Modifier and Type | Method and Description |
---|---|
private void |
populateThetas(float[] nn_params) |
double |
predict(double[] unseen) |
float |
predict(float[] unseen)
The given input vector of unseen is m=(256 + 1) * n= 1 this returns a
prediction probability
|
private int numOfInputs
private int numOfHidden
private int numOfOutputs
private float[][] Theta1
private float[][] Theta2
public NNTrainedModel(int nInput, int nHidden, int nOutput, float[] nn_params)
private void populateThetas(float[] nn_params)
public double predict(double[] unseen)
predict
in class TrainedModel
public float predict(float[] unseen)
predict
in class TrainedModel