public class SoftmaxRegression extends BayesianModel
minimize the negative ELBO ELBO = E[log(P(yIn|xIn,weights)] - D_KL[weights,prior]
BayesianModel.OptimizerType
Constructor and Description |
---|
SoftmaxRegression(int numberOfClasses) |
SoftmaxRegression(SoftmaxRegression model) |
Modifier and Type | Method and Description |
---|---|
double |
evaluate(double[] features,
Object expected) |
String |
getClassLabel(Number index) |
protected double[][] |
getUpdatedWeights()
implements the model specific gradient updates.
|
void |
setNoOfClasses(int val) |
double[] |
update(double[] features,
String target) |
getLearningRate, getNumFeatures, getNumSamples, getOptimizerType, initiateModel, predict, predictWithStd, setAddBias, setLearningRate, setNumFeatures, setNumSamples, setOptimizerType, setPredictionSamples, update
public SoftmaxRegression(int numberOfClasses)
public SoftmaxRegression(SoftmaxRegression model)
protected double[][] getUpdatedWeights()
BayesianModel
getUpdatedWeights
in class BayesianModel
public double evaluate(double[] features, Object expected)
evaluate
in class BayesianModel
public void setNoOfClasses(int val)
public double[] update(double[] features, String target)
Copyright © 2019 WSO2. All rights reserved.