nnlib
GPU-accelerated, C/C++ neural network library.
Network Member List

This is the complete list of members for Network, including all inherited members.

add(size_t numNeurons, const std::string &activation="linear")Network
forward(const sTensor &batch)Network
layersNetworkprivate
locationNetworkprivate
Network(size_t inputSize, bool useGPU=true, long long seed=NO_SEED)Networkexplicit
previousSizeNetworkprivate
processEpoch(std::vector< sTensor > &batches, std::vector< sTensor > &targets, std::vector< sTensor > &targetsOnHost, float learningRate, Loss *loss, std::vector< Metric * > &metrics)Networkprivate
seedNetworkprivate
train(sTensor &X, sTensor &y, int epochs, size_t batchSize, float learningRate, Loss *loss, std::vector< Metric * > &metrics)Network