greycat.ml.neuralnet

Class NeuralNet

java.lang.Object

greycat.ml.neuralnet.NeuralNet

public class NeuralNet
extends Object

Constructor Summary

Constructors

Constructor and Description
NeuralNet(EGraph p_backend)

Method Summary

Modifier and Type	Method and Description
NeuralNet	addLayer(int layerType, int inputs, int outputs, int activationUnit, double[] activationParams)
void	finalLearn()
DMatrix[]	learn(double[] inputs, double[] outputs, boolean reportLoss)
DMatrix[]	learnVec(DMatrix inputs, DMatrix outputs, boolean reportLoss)
double[]	predict(double[] inputs)
DMatrix	predictVec(DMatrix inputs)
void	resetState()
void	setOptimizer(int optimizer, double[] learnerParams, int frequency)
void	setRandom(long seed, double std)
void	setTestLoss(int testLoss)
void	setTrainLoss(int trainLoss)
DMatrix[]	testVec(DMatrix inputs, DMatrix outputs)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

NeuralNet

public NeuralNet(EGraph p_backend)

Method Detail

setTrainLoss

public void setTrainLoss(int trainLoss)

setTestLoss

public void setTestLoss(int testLoss)

setOptimizer

public void setOptimizer(int optimizer,
                         double[] learnerParams,
                         int frequency)

setRandom

public void setRandom(long seed,
                      double std)

addLayer

public NeuralNet addLayer(int layerType,
                          int inputs,
                          int outputs,
                          int activationUnit,
                          double[] activationParams)

learn

public DMatrix[] learn(double[] inputs,
                       double[] outputs,
                       boolean reportLoss)

learnVec

public DMatrix[] learnVec(DMatrix inputs,
                          DMatrix outputs,
                          boolean reportLoss)

testVec

public DMatrix[] testVec(DMatrix inputs,
                         DMatrix outputs)

predictVec

public DMatrix predictVec(DMatrix inputs)

finalLearn

public final void finalLearn()

resetState

public void resetState()

predict

public double[] predict(double[] inputs)