public class NNTrainedModel extends TrainedModel
Constructor and Description |
---|
NNTrainedModel(int nInput,
int nHidden,
int nOutput,
float[] nn_params) |
Modifier and Type | Method and Description |
---|---|
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
|
public NNTrainedModel(int nInput, int nHidden, int nOutput, float[] nn_params)
public double predict(double[] unseen)
predict
in class TrainedModel
public float predict(float[] unseen)
predict
in class TrainedModel
Copyright © 2007–2021 The Apache Software Foundation. All rights reserved.